edu.udo.cs.yale.operator.learner
Class DistributionModel

java.lang.Object
  edu.udo.cs.yale.operator.AbstractIOObject
      edu.udo.cs.yale.operator.ResultObjectAdapter
          edu.udo.cs.yale.operator.AbstractModel
              edu.udo.cs.yale.operator.IOModel
                  edu.udo.cs.yale.operator.learner.PredictionModel
                      edu.udo.cs.yale.operator.learner.SimplePredictionModel
                          edu.udo.cs.yale.operator.learner.DistributionModel

All Implemented Interfaces:

IOObject, Model, ResultObject, Saveable

public class DistributionModel
extends SimplePredictionModel

DistributionModel is a model for learner, which generate estimated distributions for prediction. For all classes and all attributes at least one distribution must be added. Predicted will be the class with the highest probability. The classprobability is the product of the probability of every attribute. The probability for an attribute is the mean of the probabilities of every distribution.

Version:

$Id$

Author:

Sebastian Land

Field Summary

private java.util.ArrayList<java.util.ArrayList<java.util.ArrayList<Distribution>>>	classDistributions
private double[]	classProbabilities
private int	numberOfAttributes
private int	numberOfClasses
private boolean[]	useAttribute

Constructor Summary

DistributionModel(Attribute labelAttribute, int numberOfClasses, double[] classProbabilities, boolean[] useAttribute)

Method Summary

void	addDistribution(int classIndex, int attributeIndex, Distribution distribution)
private double	getProbabilityForAttribute(int classIndex, int attributeIndex, Example example)
private double	getProbabilityForClass(int classIndex, Example example)
double	predict(Example example) Applies the model to a single example and returns the predicted class value.
void	readPredictionModelData(java.io.ObjectInputStream in) Reads the model data (without label).
java.lang.String	toResultString() The default implementation simply returns the result of the method Object.toString().
void	writePredictionModelData(java.io.ObjectOutputStream out) Writes the model data (without label).

Methods inherited from class edu.udo.cs.yale.operator.learner.SimplePredictionModel

performPrediction

Methods inherited from class edu.udo.cs.yale.operator.learner.PredictionModel

apply, createPredictedLabel, getLabel, readData, removePredictedLabel, setLabel, toString, writeData

Methods inherited from class edu.udo.cs.yale.operator.AbstractModel

equals, getExtension, getFileDescription, readModel, readModel, save, setParameter, writeModel, writeModel

Methods inherited from class edu.udo.cs.yale.operator.ResultObjectAdapter

addAction, getActions, getName, getVisualisationComponent, toHTML

Methods inherited from class edu.udo.cs.yale.operator.AbstractIOObject

copy, read, read, read, write, write

Methods inherited from class java.lang.Object

clone, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface edu.udo.cs.yale.operator.ResultObject

getActions, getName, getVisualisationComponent

Methods inherited from interface edu.udo.cs.yale.operator.IOObject

copy, write, write

Field Detail

useAttribute

private boolean[] useAttribute

classDistributions

private java.util.ArrayList<java.util.ArrayList<java.util.ArrayList<Distribution>>> classDistributions

numberOfClasses

private int numberOfClasses

numberOfAttributes

private int numberOfAttributes

classProbabilities

private double[] classProbabilities

Constructor Detail

DistributionModel

public DistributionModel(Attribute labelAttribute,
                         int numberOfClasses,
                         double[] classProbabilities,
                         boolean[] useAttribute)

Method Detail

addDistribution

public void addDistribution(int classIndex,
                            int attributeIndex,
                            Distribution distribution)

predict

public double predict(Example example)
               throws OperatorException

Description copied from class: SimplePredictionModel

Applies the model to a single example and returns the predicted class value.

Specified by:

predict in class SimplePredictionModel

Throws:

OperatorException

getProbabilityForClass

private double getProbabilityForClass(int classIndex,
                                      Example example)

getProbabilityForAttribute

private double getProbabilityForAttribute(int classIndex,
                                          int attributeIndex,
                                          Example example)

readPredictionModelData

public void readPredictionModelData(java.io.ObjectInputStream in)
                             throws java.io.IOException

Description copied from class: PredictionModel

Reads the model data (without label).

Specified by:

readPredictionModelData in class PredictionModel

Throws:

java.io.IOException

writePredictionModelData

public void writePredictionModelData(java.io.ObjectOutputStream out)
                              throws java.io.IOException

Description copied from class: PredictionModel

Writes the model data (without label).

Specified by:

writePredictionModelData in class PredictionModel

Throws:

java.io.IOException

toResultString

public java.lang.String toResultString()

Description copied from class: ResultObjectAdapter

The default implementation simply returns the result of the method Object.toString().

Specified by:

toResultString in interface ResultObject

Overrides:

toResultString in class ResultObjectAdapter