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 tests\Phpml\Metric;
|
|
|
|
|
|
|
|
use Phpml\Metric\ConfusionMatrix;
|
|
|
|
|
|
|
|
class ConfusionMatrixTest extends \PHPUnit_Framework_TestCase
|
|
|
|
{
|
|
|
|
public function testComputeConfusionMatrixOnNumericLabels()
|
|
|
|
{
|
|
|
|
$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()
|
|
|
|
{
|
|
|
|
$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()
|
|
|
|
{
|
|
|
|
$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));
|
|
|
|
}
|
|
|
|
}
|