diff --git a/lib/Service/MLP/OptimizerService.php b/lib/Service/MLP/OptimizerService.php index ec847cd5..fdf175f0 100644 --- a/lib/Service/MLP/OptimizerService.php +++ b/lib/Service/MLP/OptimizerService.php @@ -72,20 +72,20 @@ private function getAverageCost(OutputInterface $output, TrainingResult ...$results): float { $costs = array_map(function (TrainingResult $result) use ($output) { $output->writeln(sprintf(" Training result: f1=%f, p(n)=%f, r(n)=%f, f1(n)=%f, p(y)=%f, r(y)=%f, f1(y)=%f, PSR=%d/%d/%d", - $result->getReport()['overall']['f1_score'], + $result->getReport()['overall']['f1 score'], $result->getReport()['classes']['n']['precision'], $result->getReport()['classes']['n']['recall'], - $result->getReport()['classes']['n']['f1_score'], + $result->getReport()['classes']['n']['f1 score'], $result->getReport()['classes']['y']['precision'], $result->getReport()['classes']['y']['recall'], - $result->getReport()['classes']['y']['f1_score'], + $result->getReport()['classes']['y']['f1 score'], $result->getModel()->getSamplesPositive(), $result->getModel()->getSamplesShuffled(), $result->getModel()->getSamplesRandom() )); return ( - $result->getReport()['classes']['n']['f1_score'] + - $result->getReport()['overall']['f1_score'] + $result->getReport()['classes']['n']['f1 score'] + + $result->getReport()['overall']['f1 score'] ) / 2; }, $results);