ml-naivebayes

Naive bayes library

Usage no npm install needed!

<script type="module">
  import mlNaivebayes from 'https://cdn.skypack.dev/ml-naivebayes';
</script>

README

ml-naivebayes

NPM version build status Test coverage npm download

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);

Authors

License

MIT