Camspiers\StatisticalClassifier\Classifier\SVM::getProbabilities PHP Method

getProbabilities() public method

Returns the probabilities of the document being in each category
public getProbabilities ( string $document ) : array
$document string
return array
    public function getProbabilities($document)
    {
        if ($this->hasThreshold()) {
            $model = $this->preparedModel();
            $data = $this->prepareDocument($document, $model);
            $probabilities = array();
            $model->getModel()->predict_probability($data, $probabilities);
            return array_combine($model->getCategoryMap(), $probabilities);
        }
    }