MDL-58992 analytics: add multi-classification to predictors

Adds multi-class capabilities to prediction processors
as well as multi-classification unit tests
This commit is contained in:
Vlad Apetrei
2019-09-09 08:49:56 +08:00
committed by David Monllaó
parent a672f021ea
commit 8cb90066d4
5 changed files with 422 additions and 5 deletions
+2 -3
View File
@@ -132,8 +132,7 @@ class processor implements \core_analytics\classifier, \core_analytics\regressor
$nsamples = count($samples);
if ($nsamples === self::BATCH_SIZE) {
// Training it batches to avoid running out of memory.
$classifier->partialTrain($samples, $targets, array(0, 1));
$classifier->partialTrain($samples, $targets, json_decode($metadata['targetclasses']));
$samples = array();
$targets = array();
}
@@ -152,7 +151,7 @@ class processor implements \core_analytics\classifier, \core_analytics\regressor
// Train the remaining samples.
if ($samples) {
$classifier->partialTrain($samples, $targets, array(0, 1));
$classifier->partialTrain($samples, $targets, json_decode($metadata['targetclasses']));
}
$resultobj = new \stdClass();