mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-11-05 13:07:52 +00:00
58 lines
1.3 KiB
PHP
58 lines
1.3 KiB
PHP
<?php
|
|
|
|
declare (strict_types = 1);
|
|
|
|
namespace tests;
|
|
|
|
use Phpml\Classification\SVC;
|
|
use Phpml\FeatureExtraction\TfIdfTransformer;
|
|
use Phpml\Pipeline;
|
|
use Phpml\Preprocessing\Imputer;
|
|
use Phpml\Preprocessing\Normalizer;
|
|
use Phpml\Preprocessing\Imputer\Strategy\MostFrequentStrategy;
|
|
|
|
class PipelineTest extends \PHPUnit_Framework_TestCase
|
|
{
|
|
public function testPipelineConstruction()
|
|
{
|
|
$transformers = [
|
|
new TfIdfTransformer(),
|
|
];
|
|
$estimator = new SVC();
|
|
|
|
$pipeline = new Pipeline($transformers, $estimator);
|
|
|
|
$this->assertEquals($transformers, $pipeline->getTransformers());
|
|
$this->assertEquals($estimator, $pipeline->getEstimator());
|
|
}
|
|
|
|
public function testPipelineWorkflow()
|
|
{
|
|
$transformers = [
|
|
new Imputer(null, new MostFrequentStrategy()),
|
|
new Normalizer(),
|
|
];
|
|
$estimator = new SVC();
|
|
|
|
$samples = [
|
|
[1, -1, 2],
|
|
[2, 0, null],
|
|
[null, 1, -1],
|
|
];
|
|
|
|
$targets = [
|
|
4,
|
|
1,
|
|
4
|
|
];
|
|
|
|
$pipeline = new Pipeline($transformers, $estimator);
|
|
$pipeline->train($samples, $targets);
|
|
|
|
$predicted = $pipeline->predict([[0, 0, 0]]);
|
|
|
|
$this->assertEquals(4, $predicted[0]);
|
|
}
|
|
|
|
}
|