com.cloudera.oryx.app.classreg.predict

Class CategoricalPrediction

java.lang.Object

com.cloudera.oryx.app.classreg.predict.Prediction

com.cloudera.oryx.app.classreg.predict.CategoricalPrediction

All Implemented Interfaces:

Serializable

public final class CategoricalPrediction
extends Prediction

Represents a prediction of the value of a categorical target. The prediction is not just a single category value, but a probability distribution over all known category values.

See Also:

NumericPrediction, Serialized Form

Constructor Summary

Constructors

Constructor and Description
CategoricalPrediction(double[] categoryCounts)
CategoricalPrediction(int[] categoryCounts)

Method Summary

Modifier and Type	Method and Description
boolean	equals(Object o)
double[]	getCategoryCounts()
double[]	getCategoryProbabilities()
FeatureType	getFeatureType()
int	getMostProbableCategoryEncoding()
int	hashCode()
String	toString()
void	update(Example train)
void	update(int encoding, int count)

Methods inherited from class com.cloudera.oryx.app.classreg.predict.Prediction

getCount

Methods inherited from class java.lang.Object

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

Constructor Detail

CategoricalPrediction

public CategoricalPrediction(int[] categoryCounts)

CategoricalPrediction

public CategoricalPrediction(double[] categoryCounts)

Parameters:

categoryCounts - "counts" for each category, which may be fractional

Method Detail

getCategoryCounts

public double[] getCategoryCounts()

getCategoryProbabilities

public double[] getCategoryProbabilities()

getMostProbableCategoryEncoding

public int getMostProbableCategoryEncoding()

getFeatureType

public FeatureType getFeatureType()

Specified by:

getFeatureType in class Prediction

Returns:

FeatureType.CATEGORICAL

update

public void update(Example train)

Specified by:

update in class Prediction

update

public void update(int encoding,
                   int count)

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object

toString

public String toString()

Specified by:

toString in class Prediction