mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-11-26 06:46:45 +00:00
33 lines
1.1 KiB
PHP
33 lines
1.1 KiB
PHP
|
<?php
|
||
|
|
declare(strict_types = 1);
|
||
|
|
|
||
|
|
namespace tests\Phpml\Metric;
|
||
|
|
|
||
|
|
use Phpml\Metric\ClassificationReport;
|
||
|
|
|
||
|
|
class ClassificationReportTest extends \PHPUnit_Framework_TestCase
|
||
|
|
{
|
||
|
|
|
||
|
|
public function testClassificationReportGenerateWithStringLabels()
|
||
|
|
{
|
||
|
|
$labels = ['cat', 'ant', 'bird', 'bird', 'bird'];
|
||
|
|
$predicted = ['cat', 'cat', 'bird', 'bird', 'ant'];
|
||
|
|
|
||
|
|
$report = new ClassificationReport($labels, $predicted);
|
||
|
|
|
||
|
|
$precision = ['cat' => 0.5, 'ant' => 0.0, 'bird' => 1.0];
|
||
|
|
$recall = ['cat' => 1.0, 'ant' => 0.0, 'bird' => 0.67];
|
||
|
|
$f1score = ['cat' => 0.67, 'ant' => 0.0, 'bird' => 0.80];
|
||
|
|
$support = ['cat' => 1, 'ant' => 1, 'bird' => 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);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|