Phpml\Preprocessing\LabelEncoder

Complexity

Total Complexity

8

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	fit()	0	10	3
A	transform()	0	6	2
A	inverseTransform()	0	7	2
A	classes()	0	4	1

<?php

declare(strict_types=1);

namespace Phpml\Preprocessing;

final class LabelEncoder implements Preprocessor
{
    /**
     * @var int[]
     */
    private $classes = [];

    public function fit(array $samples, ?array $targets = null): void
    {
        $this->classes = [];

        foreach ($samples as $sample) {
            if (!isset($this->classes[(string) $sample])) {
                $this->classes[(string) $sample] = count($this->classes);
            }
        }
    }

    public function transform(array &$samples): void
    {
        foreach ($samples as &$sample) {
            $sample = $this->classes[(string) $sample];
        }
    }

    public function inverseTransform(array &$samples): void
    {
        $classes = array_flip($this->classes);
        foreach ($samples as &$sample) {
            $sample = $classes[$sample];
        }
    }

    /**
     * @return string[]
     */
    public function classes(): array
    {
        return array_keys($this->classes);
    }
}