mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-06-16 07:12:21 +00:00
Add test for Pipeline save and restore with ModelManager (#191)
This commit is contained in:
parent
d953ef6bfc
commit
7435bece34
|
@ -7,6 +7,7 @@ namespace Phpml\Tests;
|
||||||
use Phpml\Classification\SVC;
|
use Phpml\Classification\SVC;
|
||||||
use Phpml\FeatureExtraction\TfIdfTransformer;
|
use Phpml\FeatureExtraction\TfIdfTransformer;
|
||||||
use Phpml\FeatureExtraction\TokenCountVectorizer;
|
use Phpml\FeatureExtraction\TokenCountVectorizer;
|
||||||
|
use Phpml\ModelManager;
|
||||||
use Phpml\Pipeline;
|
use Phpml\Pipeline;
|
||||||
use Phpml\Preprocessing\Imputer;
|
use Phpml\Preprocessing\Imputer;
|
||||||
use Phpml\Preprocessing\Imputer\Strategy\MostFrequentStrategy;
|
use Phpml\Preprocessing\Imputer\Strategy\MostFrequentStrategy;
|
||||||
|
@ -104,4 +105,40 @@ class PipelineTest extends TestCase
|
||||||
|
|
||||||
$this->assertEquals($expected, $predicted);
|
$this->assertEquals($expected, $predicted);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public function testSaveAndRestore(): void
|
||||||
|
{
|
||||||
|
$pipeline = new Pipeline([
|
||||||
|
new TokenCountVectorizer(new WordTokenizer()),
|
||||||
|
new TfIdfTransformer(),
|
||||||
|
], new SVC());
|
||||||
|
|
||||||
|
$pipeline->train([
|
||||||
|
'Hello Paul',
|
||||||
|
'Hello Martin',
|
||||||
|
'Goodbye Tom',
|
||||||
|
'Hello John',
|
||||||
|
'Goodbye Alex',
|
||||||
|
'Bye Tony',
|
||||||
|
], [
|
||||||
|
'greetings',
|
||||||
|
'greetings',
|
||||||
|
'farewell',
|
||||||
|
'greetings',
|
||||||
|
'farewell',
|
||||||
|
'farewell',
|
||||||
|
]);
|
||||||
|
|
||||||
|
$testSamples = ['Hello Max', 'Goodbye Mark'];
|
||||||
|
$predicted = $pipeline->predict($testSamples);
|
||||||
|
|
||||||
|
$filepath = tempnam(sys_get_temp_dir(), uniqid('pipeline-test', true));
|
||||||
|
$modelManager = new ModelManager();
|
||||||
|
$modelManager->saveToFile($pipeline, $filepath);
|
||||||
|
|
||||||
|
$restoredClassifier = $modelManager->restoreFromFile($filepath);
|
||||||
|
$this->assertEquals($pipeline, $restoredClassifier);
|
||||||
|
$this->assertEquals($predicted, $restoredClassifier->predict($testSamples));
|
||||||
|
unlink($filepath);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user