nlp/com.londogard.nlp.meachinelearning.predictors.classifiers/NaiveBayes

NaiveBayes

public final class NaiveBayes implements Classifier

A Naïve Bayes Classifier that is using Laplacian Smoothing.

Constructors

public NaiveBayes NaiveBayes()

Functions

public Unit fit(MultiArray<Float, D2> X, NDArray<Integer, D2> y)

public final NDArray<Float, D2> getLogLambda()

public final Float getLogPrior()

public NDArray<Integer, D2> predict(MultiArray<Float, D2> X)

public final NDArray<Float, D2> predictProba(NDArray<Float, D2> X)

public final Unit setLogLambda(NDArray<Float, D2> logLambda)

public final Unit setLogPrior(Float logPrior)

Properties

private NDArray<Float, D2> logLambda

private Float logPrior