mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-06-06 10:30:48 +00:00
726cf4cddf
* travis: move coveralls here, decouple from package * composer: use PSR4 * phpunit: simpler config * travis: add ecs run * composer: add ecs dev * use standard vendor/bin directory for dependency bins, confuses with local bins and require gitignore handling * ecs: add PSR2 * [cs] PSR2 spacing fixes * [cs] PSR2 class name fix * [cs] PHP7 fixes - return semicolon spaces, old rand functions, typehints * [cs] fix less strict typehints * fix typehints to make tests pass * ecs: ignore typehint-less elements * [cs] standardize arrays * [cs] standardize docblock, remove unused comments * [cs] use self where possible * [cs] sort class elements, from public to private * [cs] do not use yoda (found less yoda-cases, than non-yoda) * space * [cs] do not assign in condition * [cs] use namespace imports if possible * [cs] use ::class over strings * [cs] fix defaults for arrays properties, properties and constants single spacing * cleanup ecs comments * [cs] use item per line in multi-items array * missing line * misc * rebase
84 lines
1.7 KiB
PHP
84 lines
1.7 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Phpml\Preprocessing;
|
|
|
|
use Phpml\Preprocessing\Imputer\Strategy;
|
|
|
|
class Imputer implements Preprocessor
|
|
{
|
|
public const AXIS_COLUMN = 0;
|
|
|
|
public const AXIS_ROW = 1;
|
|
|
|
/**
|
|
* @var mixed
|
|
*/
|
|
private $missingValue;
|
|
|
|
/**
|
|
* @var Strategy
|
|
*/
|
|
private $strategy;
|
|
|
|
/**
|
|
* @var int
|
|
*/
|
|
private $axis;
|
|
|
|
/**
|
|
* @var
|
|
*/
|
|
private $samples;
|
|
|
|
/**
|
|
* @param mixed $missingValue
|
|
* @param array|null $samples
|
|
*/
|
|
public function __construct($missingValue, Strategy $strategy, int $axis = self::AXIS_COLUMN, array $samples = [])
|
|
{
|
|
$this->missingValue = $missingValue;
|
|
$this->strategy = $strategy;
|
|
$this->axis = $axis;
|
|
$this->samples = $samples;
|
|
}
|
|
|
|
public function fit(array $samples): void
|
|
{
|
|
$this->samples = $samples;
|
|
}
|
|
|
|
public function transform(array &$samples): void
|
|
{
|
|
foreach ($samples as &$sample) {
|
|
$this->preprocessSample($sample);
|
|
}
|
|
}
|
|
|
|
private function preprocessSample(array &$sample): void
|
|
{
|
|
foreach ($sample as $column => &$value) {
|
|
if ($value === $this->missingValue) {
|
|
$value = $this->strategy->replaceValue($this->getAxis($column, $sample));
|
|
}
|
|
}
|
|
}
|
|
|
|
private function getAxis(int $column, array $currentSample): array
|
|
{
|
|
if ($this->axis === self::AXIS_ROW) {
|
|
return array_diff($currentSample, [$this->missingValue]);
|
|
}
|
|
|
|
$axis = [];
|
|
foreach ($this->samples as $sample) {
|
|
if ($sample[$column] !== $this->missingValue) {
|
|
$axis[] = $sample[$column];
|
|
}
|
|
}
|
|
|
|
return $axis;
|
|
}
|
|
}
|