octoleo/php-ml
1
0
Fork 0
mirror of https://github.com/Llewellynvdm/php-ml.git synced 2025-01-24 23:58:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

add more tests for CReport

Browse Source
This commit is contained in:
Arkadiusz Kondas 2016-07-19 22:01:39 +02:00
parent 074dcf7470
commit 093e8fc89c
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/Phpml/Metric/ClassificationReportTest.php
View File

@ -27,4 +27,24 @@ class ClassificationReportTest extends \PHPUnit_Framework_TestCase
$this->assertEquals($support, $report->getSupport(), '', 0.01);
$this->assertEquals($average, $report->getAverage(), '', 0.01);
}
public function testClassificationReportGenerateWithNumericLabels()
{
$labels = [0, 1, 2, 2, 2];
$predicted = [0, 0, 2, 2, 1];
$report = new ClassificationReport($labels, $predicted);
$precision = [0 => 0.5, 1 => 0.0, 2 => 1.0];
$recall = [0 => 1.0, 1 => 0.0, 2 => 0.67];
$f1score = [0 => 0.67, 1 => 0.0, 2 => 0.80];
$support = [0 => 1, 1 => 1, 2 => 3];
$average = ['precision' => 0.75, 'recall' => 0.83, 'f1score' => 0.73];
$this->assertEquals($precision, $report->getPrecision(), '', 0.01);
$this->assertEquals($recall, $report->getRecall(), '', 0.01);
$this->assertEquals($f1score, $report->getF1score(), '', 0.01);
$this->assertEquals($support, $report->getSupport(), '', 0.01);
$this->assertEquals($average, $report->getAverage(), '', 0.01);
}
}