org.apache.ignite.ml.naivebayes.gaussian

Class GaussianNaiveBayesModel

java.lang.Object

org.apache.ignite.ml.naivebayes.gaussian.GaussianNaiveBayesModel

All Implemented Interfaces:

Serializable, AutoCloseable, Exportable<GaussianNaiveBayesModel>, IgniteModel<Vector,Double>, Model<Vector,Double>

public final class GaussianNaiveBayesModel
extends Object
implements IgniteModel<Vector,Double>, Exportable<GaussianNaiveBayesModel>, Serializable

Simple naive Bayes model which predicts result value y belongs to a class C_k, k in [0..K] as p(C_k,y) = p(C_k)*p(y_1,C_k) *...*p(y_n,C_k) / p(y). Return the number of the most possible class.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GaussianNaiveBayesModel(double[][] means, double[][] variances, double[] classProbabilities, double[] labels, org.apache.ignite.ml.naivebayes.gaussian.GaussianNaiveBayesSumsHolder sumsHolder)

Method Summary

Modifier and Type	Method and Description
double[]	getClassProbabilities()
double[][]	getMeans()
org.apache.ignite.ml.naivebayes.gaussian.GaussianNaiveBayesSumsHolder	getSumsHolder()
double[][]	getVariances()
Double	predict(Vector vector) Returns a number of class to which the input belongs.
<P> void	saveModel(Exporter<GaussianNaiveBayesModel,P> exporter, P path) Save model by the given path.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.ignite.ml.IgniteModel

andBefore, andThen, andThen, close, combine, toString

Constructor Detail

GaussianNaiveBayesModel

public GaussianNaiveBayesModel(double[][] means,
                               double[][] variances,
                               double[] classProbabilities,
                               double[] labels,
                               org.apache.ignite.ml.naivebayes.gaussian.GaussianNaiveBayesSumsHolder sumsHolder)

Parameters:

means - Means of features for all classes.

variances - Variances of features for all classes.

classProbabilities - Probabilities for all classes.

labels - Labels.

sumsHolder - Feature sum, squared sum and count sum per label. This data is used for future model updating.

Method Detail

saveModel

public <P> void saveModel(Exporter<GaussianNaiveBayesModel,P> exporter,
                          P path)

Save model by the given path.

Specified by:

saveModel in interface Exportable<GaussianNaiveBayesModel>

Parameters:

exporter - Exporter.

path - Path to saved model.

predict

public Double predict(Vector vector)

Returns a number of class to which the input belongs.

Specified by:

predict in interface Model<Vector,Double>

Parameters:

vector - Input arguments.

Returns:

Prediction result.

getMeans

public double[][] getMeans()

getVariances

public double[][] getVariances()

getClassProbabilities

public double[] getClassProbabilities()

getSumsHolder

public org.apache.ignite.ml.naivebayes.gaussian.GaussianNaiveBayesSumsHolder getSumsHolder()