Package Exports
- ml-naivebayes
This package does not declare an exports field, so the exports above have been automatically detected and optimized by JSPM instead. If any package subpath is missing, it is recommended to post an issue to the original package (ml-naivebayes) to support the "exports" field. If that is not possible, create a JSPM override to customize the exports field for this package.
Readme
Naive Bayes
Naive bayes classifier.
Methods
new NaiveBayes()
Constructor that takes no arguments.
Example
var nb = new NaiveBayes();train(trainingSet, predictions)
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(dataset)
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);export()
Exports the actual Naive Bayes model to an Javascript Object.
load(model)
Returns a new Naive Bayes Classifier with the given model.
Arguments
model- Javascript Object generated from export() function.