php-ml/tests/Metric/ConfusionMatrixTest.php

63 lines
1.7 KiB
PHP
Raw Normal View History

2016-07-06 22:29:58 +00:00
<?php
2016-11-20 21:53:17 +00:00
declare(strict_types=1);
2016-07-06 22:29:58 +00:00
namespace Phpml\Tests\Metric;
2016-07-06 22:29:58 +00:00
use Phpml\Metric\ConfusionMatrix;
2017-02-03 11:58:25 +00:00
use PHPUnit\Framework\TestCase;
2016-07-06 22:29:58 +00:00
2017-02-03 11:58:25 +00:00
class ConfusionMatrixTest extends TestCase
2016-07-06 22:29:58 +00:00
{
public function testComputeConfusionMatrixOnNumericLabels(): void
2016-07-06 22:29:58 +00:00
{
$actualLabels = [2, 0, 2, 2, 0, 1];
$predictedLabels = [0, 0, 2, 2, 0, 2];
$confusionMatrix = [
[2, 0, 0],
[0, 0, 1],
[1, 0, 2],
];
$this->assertEquals($confusionMatrix, ConfusionMatrix::compute($actualLabels, $predictedLabels));
}
public function testComputeConfusionMatrixOnStringLabels(): void
2016-07-06 22:29:58 +00:00
{
$actualLabels = ['cat', 'ant', 'cat', 'cat', 'ant', 'bird'];
$predictedLabels = ['ant', 'ant', 'cat', 'cat', 'ant', 'cat'];
$confusionMatrix = [
[2, 0, 0],
[0, 0, 1],
[1, 0, 2],
];
$this->assertEquals($confusionMatrix, ConfusionMatrix::compute($actualLabels, $predictedLabels));
}
public function testComputeConfusionMatrixOnLabelsWithSubset(): void
2016-07-06 22:29:58 +00:00
{
$actualLabels = ['cat', 'ant', 'cat', 'cat', 'ant', 'bird'];
$predictedLabels = ['ant', 'ant', 'cat', 'cat', 'ant', 'cat'];
$labels = ['ant', 'bird'];
$confusionMatrix = [
[2, 0],
[0, 0],
];
$this->assertEquals($confusionMatrix, ConfusionMatrix::compute($actualLabels, $predictedLabels, $labels));
$labels = ['bird', 'ant'];
$confusionMatrix = [
[0, 0],
[0, 2],
];
$this->assertEquals($confusionMatrix, ConfusionMatrix::compute($actualLabels, $predictedLabels, $labels));
}
}