48 lines
1005 B
PHP
48 lines
1005 B
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Phpml\Dataset;
|
|
|
|
use Phpml\Exception\DatasetException;
|
|
|
|
class FilesDataset extends ArrayDataset
|
|
{
|
|
/**
|
|
* @param string $rootPath
|
|
*
|
|
* @throws DatasetException
|
|
*/
|
|
public function __construct(string $rootPath)
|
|
{
|
|
if (!is_dir($rootPath)) {
|
|
throw DatasetException::missingFolder($rootPath);
|
|
}
|
|
|
|
$this->scanRootPath($rootPath);
|
|
}
|
|
|
|
/**
|
|
* @param string $rootPath
|
|
*/
|
|
private function scanRootPath(string $rootPath)
|
|
{
|
|
foreach (glob($rootPath.DIRECTORY_SEPARATOR.'*', GLOB_ONLYDIR) as $dir) {
|
|
$this->scanDir($dir);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param string $dir
|
|
*/
|
|
private function scanDir(string $dir)
|
|
{
|
|
$target = basename($dir);
|
|
|
|
foreach (array_filter(glob($dir.DIRECTORY_SEPARATOR.'*'), 'is_file') as $file) {
|
|
$this->samples[] = [file_get_contents($file)];
|
|
$this->targets[] = $target;
|
|
}
|
|
}
|
|
}
|