Naive bayes classifier.

Constructor that takes no arguments.

Example

var nb = new NaiveBayes();

Train the Naive Bayes model to the given training set and predictions

Arguments

• trainingSet - A matrix of the training set.
• trainingLabels - An array of value for each case in the training set.

Example

var cases = [[6,148,72,35,0,33.6,0.627,5],
             [1.50,85,66.5,29,0,26.6,0.351,31],
             [8,183,64,0,0,23.3,0.672,32],
             [0.5,89,65.5,23,94,28.1,0.167,21],
             [0,137,40,35,168,43.1,2.288,33]];
var predictions = [1, 0, 1, 0, 1];

nb.train(trainingSet, predictions);

Predict the values of the dataset.

Arguments

• dataset - A matrix that contains the dataset.

Example

var dataset = [[6,148,72,35,0,33.6,0.627,5],
               [1.50,85,66.5,29,0,26.6,0.351,31]];

var ans = nb.predict(dataset);

Exports the actual Naive Bayes model to an Javascript Object.

Returns a new Naive Bayes Classifier with the given model.

Arguments

• model - Javascript Object generated from export() function.

• Jefferson Hernandez

MIT