mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-12-02 01:48:22 +00:00
28 lines
609 B
Markdown
28 lines
609 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`):
|
||
|
|
|
||
|
|
```
|
||
|
|
$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']
|
||
|
|
```
|