67 lines
1.8 KiB
PHP
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);
|
|
}
|
|
}
|