ml-naivebayes
v4.0.0
Published
Naive bayes library
Downloads
983
Readme
ml-naivebayes
Naive bayes classifiers.
Installation
npm install ml-naivebayes
API Documentation
Usage
GaussianNB
// assuming that you created Xtrain, Xtest, Ytrain, Ytest
import { GaussianNB } from 'ml-naivebayes';
var model = new GaussianNB();
model.train(Xtrain, Ytrain);
var predictions = model.predict(Xtest);
MultinomialNB
// assuming that you created Xtrain, Xtest, Ytrain, Ytest
import { MultinomialNB } from 'ml-naivebayes';
var model = new MultinomialNB();
model.train(Xtrain, Ytrain);
var predictions = model.predict(Xtest);