mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-11-22 21:15:10 +00:00
150 lines
3.7 KiB
PHP
150 lines
3.7 KiB
PHP
<?php
|
|
|
|
declare (strict_types = 1);
|
|
|
|
namespace tests\Preprocessing;
|
|
|
|
use Phpml\Preprocessing\Imputer;
|
|
use Phpml\Preprocessing\Imputer\Strategy\MeanStrategy;
|
|
use Phpml\Preprocessing\Imputer\Strategy\MedianStrategy;
|
|
use Phpml\Preprocessing\Imputer\Strategy\MostFrequentStrategy;
|
|
|
|
class ImputerTest extends \PHPUnit_Framework_TestCase
|
|
{
|
|
public function testComplementsMissingValuesWithMeanStrategyOnColumnAxis()
|
|
{
|
|
$data = [
|
|
[1, null, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[null, 6, 7, 8],
|
|
[8, 7, null, 5],
|
|
];
|
|
|
|
$imputeData = [
|
|
[1, 5.33, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[4.33, 6, 7, 8],
|
|
[8, 7, 4, 5],
|
|
];
|
|
|
|
$imputer = new Imputer(null, new MeanStrategy(), Imputer::AXIS_COLUMN);
|
|
$imputer->preprocess($data);
|
|
|
|
$this->assertEquals($imputeData, $data, '', $delta = 0.01);
|
|
}
|
|
|
|
public function testComplementsMissingValuesWithMeanStrategyOnRowAxis()
|
|
{
|
|
$data = [
|
|
[1, null, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[null, 6, 7, 8],
|
|
[8, 7, null, 5],
|
|
];
|
|
|
|
$imputeData = [
|
|
[1, 2.66, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[7, 6, 7, 8],
|
|
[8, 7, 6.66, 5],
|
|
];
|
|
|
|
$imputer = new Imputer(null, new MeanStrategy(), Imputer::AXIS_ROW);
|
|
$imputer->preprocess($data);
|
|
|
|
$this->assertEquals($imputeData, $data, '', $delta = 0.01);
|
|
}
|
|
|
|
public function testComplementsMissingValuesWithMediaStrategyOnColumnAxis()
|
|
{
|
|
$data = [
|
|
[1, null, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[null, 6, 7, 8],
|
|
[8, 7, null, 5],
|
|
];
|
|
|
|
$imputeData = [
|
|
[1, 6, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[4, 6, 7, 8],
|
|
[8, 7, 3, 5],
|
|
];
|
|
|
|
$imputer = new Imputer(null, new MedianStrategy(), Imputer::AXIS_COLUMN);
|
|
$imputer->preprocess($data);
|
|
|
|
$this->assertEquals($imputeData, $data, '', $delta = 0.01);
|
|
}
|
|
|
|
public function testComplementsMissingValuesWithMediaStrategyOnRowAxis()
|
|
{
|
|
$data = [
|
|
[1, null, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[null, 6, 7, 8],
|
|
[8, 7, null, 5],
|
|
];
|
|
|
|
$imputeData = [
|
|
[1, 3, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[7, 6, 7, 8],
|
|
[8, 7, 7, 5],
|
|
];
|
|
|
|
$imputer = new Imputer(null, new MedianStrategy(), Imputer::AXIS_ROW);
|
|
$imputer->preprocess($data);
|
|
|
|
$this->assertEquals($imputeData, $data, '', $delta = 0.01);
|
|
}
|
|
|
|
public function testComplementsMissingValuesWithMostFrequentStrategyOnColumnAxis()
|
|
{
|
|
$data = [
|
|
[1, null, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[null, 6, 7, 8],
|
|
[8, 7, null, 5],
|
|
[8, 3, 2, 5],
|
|
];
|
|
|
|
$imputeData = [
|
|
[1, 3, 3, 4],
|
|
[4, 3, 2, 1],
|
|
[8, 6, 7, 8],
|
|
[8, 7, 2, 5],
|
|
[8, 3, 2, 5],
|
|
];
|
|
|
|
$imputer = new Imputer(null, new MostFrequentStrategy(), Imputer::AXIS_COLUMN);
|
|
$imputer->preprocess($data);
|
|
|
|
$this->assertEquals($imputeData, $data);
|
|
}
|
|
|
|
public function testComplementsMissingValuesWithMostFrequentStrategyOnRowAxis()
|
|
{
|
|
$data = [
|
|
[1, null, 3, 4, 3],
|
|
[4, 3, 2, 1, 7],
|
|
[null, 6, 7, 8, 6],
|
|
[8, 7, null, 5, 5],
|
|
[8, 3, 2, 5, 4],
|
|
];
|
|
|
|
$imputeData = [
|
|
[1, 3, 3, 4, 3],
|
|
[4, 3, 2, 1, 7],
|
|
[6, 6, 7, 8, 6],
|
|
[8, 7, 5, 5, 5],
|
|
[8, 3, 2, 5, 4],
|
|
];
|
|
|
|
$imputer = new Imputer(null, new MostFrequentStrategy(), Imputer::AXIS_ROW);
|
|
$imputer->preprocess($data);
|
|
|
|
$this->assertEquals($imputeData, $data);
|
|
}
|
|
}
|