php-ml/tests/FeatureExtraction/TfIdfTransformerTest.php
2018-02-10 12:08:58 +01:00

60 lines
1.2 KiB
PHP

<?php
declare(strict_types=1);
namespace Phpml\Tests\FeatureExtraction;
use Phpml\FeatureExtraction\TfIdfTransformer;
use PHPUnit\Framework\TestCase;
class TfIdfTransformerTest extends TestCase
{
public function testTfIdfTransformation(): void
{
// https://en.wikipedia.org/wiki/Tf-idf
$samples = [
[
0 => 1,
1 => 1,
2 => 2,
3 => 1,
4 => 0,
5 => 0,
],
[
0 => 1,
1 => 1,
2 => 0,
3 => 0,
4 => 2,
5 => 3,
],
];
$tfIdfSamples = [
[
0 => 0,
1 => 0,
2 => 0.602,
3 => 0.301,
4 => 0,
5 => 0,
],
[
0 => 0,
1 => 0,
2 => 0,
3 => 0,
4 => 0.602,
5 => 0.903,
],
];
$transformer = new TfIdfTransformer($samples);
$transformer->transform($samples);
$this->assertEquals($tfIdfSamples, $samples, '', 0.001);
}
}