Estimator#
The Estimator interface is implemented by all learners in Rubix ML. It provides basic inference functionality through the predict()
method which returns a set of predictions from a dataset. Additionally, it provides methods for returning estimator type and data type compatibility declarations.
Make Predictions#
Return the predictions from a dataset containing unknown samples in an array:
public predict(Dataset $dataset) : array
$predictions = $estimator->predict($dataset);
print_r($predictions);
Array
(
[0] => married
[1] => divorced
[2] => divorced
[3] => married
)
Note
The return value of predict()
is an array containing the predictions in the same order that they were indexed in the dataset.
Last update:
2021-06-06