public function classify($document)
{
/** @var SVMModel $model */
$model = $this->preparedModel();
$categoryMap = $model->getCategoryMap();
$data = $this->prepareDocument($document, $model);
if ($this->hasThreshold()) {
$probabilities = array();
$category = $model->getModel()->predict_probability($data, $probabilities);
return $probabilities[$category] > $this->threshold ? $categoryMap[$category] : false;
} else {
$category = $model->getModel()->predict($data);
return $categoryMap[$category];
}
}