php-ml/docs/machine-learning/classification/naivebayes.md
2016-04-16 21:41:37 +02:00

609 B

NaiveBayes Classifier

Classifier based on applying Bayes' theorem with strong (naive) independence assumptions between the features.

Train

To train a classifier simply provide train samples and labels (as array):

$samples = [[5, 1, 1], [1, 5, 1], [1, 1, 5]];
$labels = ['a', 'b', 'c'];

$classifier = new NaiveBayes();
$classifier->train($samples, $labels);

Predict

To predict sample class use predict method. You can provide one sample or array of samples:

$classifier->predict([3, 1, 1]);
// return 'a'

$classifier->predict([[3, 1, 1], [1, 4, 1]);
// return ['a', 'b']