deepbox/ml
Naive Bayes
Gaussian, multinomial, Bernoulli, complement, and categorical Naive Bayes estimators.
Probabilistic
GaussianNB
Gaussian Naive Bayes classifier.
BernoulliNB
Bernoulli Naive Bayes classifier.
CategoricalNB
Categorical Naive Bayes classifier.
ComplementNB
Complement Naive Bayes classifier.
MultinomialNB
Multinomial Naive Bayes classifier.
ml-naive-bayes.ts
import { BernoulliNB, CategoricalNB, GaussianNB, MultinomialNB,} from "deepbox/ml";import { tensor } from "deepbox/ndarray";console.log(new GaussianNB().fit(tensor([[1], [2], [3], [4]]), tensor([0, 0, 1, 1])).predict(tensor([[2.5]])).toString());console.log(new BernoulliNB().fit(tensor([[0, 1], [1, 0], [1, 1]]), tensor([0, 1, 1])).predict(tensor([[1, 0]])).toString());console.log(new MultinomialNB().fit(tensor([[2, 1], [0, 3], [1, 2]]), tensor([0, 1, 1])).predict(tensor([[1, 1]])).toString());console.log(new CategoricalNB().fit(tensor([[0, 1], [1, 0], [1, 1]]), tensor([0, 1, 1])).predict(tensor([[1, 1]])).toString());