PHP Phpml\Dataset FilesDataset Examples

Example #1

File: FilesDatasetTest.php Project: php-ai/php-ml

 public function testLoadFilesDatasetWithBBCData()
 {
     $rootPath = dirname(__FILE__) . '/Resources/bbc';
     $dataset = new FilesDataset($rootPath);
     $this->assertEquals(50, count($dataset->getSamples()));
     $this->assertEquals(50, count($dataset->getTargets()));
     $targets = ['business', 'entertainment', 'politics', 'sport', 'tech'];
     $this->assertEquals($targets, array_values(array_unique($dataset->getTargets())));
     $firstSample = file_get_contents($rootPath . '/business/001.txt');
     $this->assertEquals($firstSample, $dataset->getSamples()[0][0]);
     $firstTarget = 'business';
     $this->assertEquals($firstTarget, $dataset->getTargets()[0]);
     $lastSample = file_get_contents($rootPath . '/tech/010.txt');
     $this->assertEquals($lastSample, $dataset->getSamples()[49][0]);
     $lastTarget = 'tech';
     $this->assertEquals($lastTarget, $dataset->getTargets()[49]);
 }