php-ml/tests/Preprocessing/OneHotEncoderTest.php
2019-05-15 08:00:46 +02:00

67 lines
1.8 KiB
PHP

<?php
declare(strict_types=1);
namespace Phpml\Tests\Preprocessing;
use Phpml\Exception\InvalidArgumentException;
use Phpml\Preprocessing\OneHotEncoder;
use PHPUnit\Framework\TestCase;
final class OneHotEncoderTest extends TestCase
{
public function testOneHotEncodingWithoutIgnoreUnknown(): void
{
$samples = [
['fish', 'New York', 'regression'],
['dog', 'New York', 'regression'],
['fish', 'Vancouver', 'classification'],
['dog', 'Vancouver', 'regression'],
];
$encoder = new OneHotEncoder();
$encoder->fit($samples);
$encoder->transform($samples);
self::assertEquals([
[1, 0, 1, 0, 1, 0],
[0, 1, 1, 0, 1, 0],
[1, 0, 0, 1, 0, 1],
[0, 1, 0, 1, 1, 0],
], $samples);
}
public function testThrowExceptionWhenUnknownCategory(): void
{
$encoder = new OneHotEncoder();
$encoder->fit([
['fish', 'New York', 'regression'],
['dog', 'New York', 'regression'],
['fish', 'Vancouver', 'classification'],
['dog', 'Vancouver', 'regression'],
]);
$samples = [['fish', 'New York', 'ka boom']];
$this->expectException(InvalidArgumentException::class);
$encoder->transform($samples);
}
public function testIgnoreMissingCategory(): void
{
$encoder = new OneHotEncoder(true);
$encoder->fit([
['fish', 'New York', 'regression'],
['dog', 'New York', 'regression'],
['fish', 'Vancouver', 'classification'],
['dog', 'Vancouver', 'regression'],
]);
$samples = [['ka', 'boom', 'riko']];
$encoder->transform($samples);
self::assertEquals([
[0, 0, 0, 0, 0, 0],
], $samples);
}
}