php-ml/tests/Metric/AccuracyTest.php

57 lines
1.6 KiB
PHP
Raw Normal View History

2016-04-08 20:11:59 +00:00
<?php
2016-04-08 20:49:17 +00:00
2016-11-20 21:53:17 +00:00
declare(strict_types=1);
2016-04-08 20:11:59 +00:00
namespace Phpml\Tests\Metric;
2016-04-08 20:11:59 +00:00
2016-05-31 18:01:54 +00:00
use Phpml\Classification\SVC;
use Phpml\CrossValidation\RandomSplit;
use Phpml\Dataset\Demo\IrisDataset;
use Phpml\Exception\InvalidArgumentException;
2016-04-08 20:11:59 +00:00
use Phpml\Metric\Accuracy;
2016-05-31 18:01:54 +00:00
use Phpml\SupportVectorMachine\Kernel;
2017-02-03 11:58:25 +00:00
use PHPUnit\Framework\TestCase;
2016-04-08 20:11:59 +00:00
2017-02-03 11:58:25 +00:00
class AccuracyTest extends TestCase
2016-04-08 20:11:59 +00:00
{
public function testThrowExceptionOnInvalidArguments(): void
2016-04-08 20:11:59 +00:00
{
$this->expectException(InvalidArgumentException::class);
2016-04-08 20:11:59 +00:00
$actualLabels = ['a', 'b', 'a', 'b'];
$predictedLabels = ['a', 'a'];
Accuracy::score($actualLabels, $predictedLabels);
}
public function testCalculateNormalizedScore(): void
2016-04-08 20:11:59 +00:00
{
$actualLabels = ['a', 'b', 'a', 'b'];
$predictedLabels = ['a', 'a', 'b', 'b'];
2018-10-28 06:44:52 +00:00
self::assertEquals(0.5, Accuracy::score($actualLabels, $predictedLabels));
2016-04-08 20:11:59 +00:00
}
public function testCalculateNotNormalizedScore(): void
2016-04-08 20:11:59 +00:00
{
$actualLabels = ['a', 'b', 'a', 'b'];
$predictedLabels = ['a', 'b', 'b', 'b'];
2018-10-28 06:44:52 +00:00
self::assertEquals(3, Accuracy::score($actualLabels, $predictedLabels, false));
2016-04-08 20:11:59 +00:00
}
2016-05-31 18:01:54 +00:00
public function testAccuracyOnDemoDataset(): void
2016-05-31 18:01:54 +00:00
{
$dataset = new RandomSplit(new IrisDataset(), 0.5, 123);
2016-05-31 18:01:54 +00:00
$classifier = new SVC(Kernel::RBF);
$classifier->train($dataset->getTrainSamples(), $dataset->getTrainLabels());
$predicted = (array) $classifier->predict($dataset->getTestSamples());
2016-05-31 18:01:54 +00:00
$accuracy = Accuracy::score($dataset->getTestLabels(), $predicted);
2016-12-12 18:28:26 +00:00
$expected = PHP_VERSION_ID >= 70100 ? 1 : 0.959;
self::assertEqualsWithDelta($expected, $accuracy, 0.01);
2016-05-31 18:01:54 +00:00
}
2016-04-08 20:11:59 +00:00
}