mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-11-29 00:06:31 +00:00
28 lines
618 B
Markdown
28 lines
618 B
Markdown
# 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`). Example:
|
|
|
|
```
|
|
$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 label 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']
|
|
```
|