mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-09-26 14:09:04 +00:00
37 lines
820 B
PHP
37 lines
820 B
PHP
<?php
|
|
|
|
declare (strict_types = 1);
|
|
|
|
namespace Phpml\Classification;
|
|
|
|
use Phpml\Helper\Predictable;
|
|
use Phpml\Helper\Trainable;
|
|
|
|
class NaiveBayes implements Classifier
|
|
{
|
|
use Trainable, Predictable;
|
|
|
|
/**
|
|
* @param array $sample
|
|
*
|
|
* @return mixed
|
|
*/
|
|
protected function predictSample(array $sample)
|
|
{
|
|
$predictions = [];
|
|
foreach ($this->targets as $index => $label) {
|
|
$predictions[$label] = 0;
|
|
foreach ($sample as $token => $count) {
|
|
if (array_key_exists($token, $this->samples[$index])) {
|
|
$predictions[$label] += $count * $this->samples[$index][$token];
|
|
}
|
|
}
|
|
}
|
|
|
|
arsort($predictions, SORT_NUMERIC);
|
|
reset($predictions);
|
|
|
|
return key($predictions);
|
|
}
|
|
}
|