octoleo
/
php-ml
mirror of https://github.com/Llewellynvdm/php-ml.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add delimiter option for CsvDataset (#66) 

Useful option when the CSV file uses another delimiter character other than the comma, for example, as the semicolon or tab character.
This commit is contained in:
Humberto Castelo Branco 2017-03-29 07:58:12 -03:00 committed by Arkadiusz Kondas
parent 49234429f0
commit b27f08f420
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/Phpml/Dataset/CsvDataset.php
View File

@ -20,7 +20,7 @@ class CsvDataset extends ArrayDataset
*
* @throws FileException
*/
public function __construct(string $filepath, int $features, bool $headingRow = true)
public function __construct(string $filepath, int $features, bool $headingRow = true, string $delimiter = ',')
{
if (!file_exists($filepath)) {
throw FileException::missingFile(basename($filepath));
@ -31,13 +31,13 @@ class CsvDataset extends ArrayDataset
}
if ($headingRow) {
$data = fgetcsv($handle, 1000, ',');
$data = fgetcsv($handle, 1000, $delimiter);
$this->columnNames = array_slice($data, 0, $features);
} else {
$this->columnNames = range(0, $features - 1);
}
while (($data = fgetcsv($handle, 1000, ',')) !== false) {
while (($data = fgetcsv($handle, 1000, $delimiter)) !== false) {
$this->samples[] = array_slice($data, 0, $features);
$this->targets[] = $data[$features];
}