| protected predictSample ( array $sample ) : mixed | ||
| $sample | array | |
| 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);
}