Phpml\Preprocessing\ColumnFilter

Complexity

Total Complexity

4

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	fit()	0	2	1
A	transform()	0	6	2
A	__construct()	0	8	1

<?php

declare(strict_types=1);

namespace Phpml\Preprocessing;

final class ColumnFilter implements Preprocessor
{
    /**
     * @var string[]
     */
    private $datasetColumns = [];

    /**
     * @var string[]
     */
    private $filterColumns = [];

    public function __construct(array $datasetColumns, array $filterColumns)
    {
        $this->datasetColumns = array_map(static function (string $column): string {
            return $column;
        }, $datasetColumns);
        $this->filterColumns = array_map(static function (string $column): string {
            return $column;
        }, $filterColumns);
    }

    public function fit(array $samples, ?array $targets = null): void
    {
        //nothing to do
    }

    public function transform(array &$samples, ?array &$targets = null): void
    {
        $keys = array_intersect($this->datasetColumns, $this->filterColumns);

        foreach ($samples as &$sample) {
            $sample = array_values(array_intersect_key($sample, $keys));
        }
    }
}