Phpml\Metric\ConfusionMatrix::compute()

Complexity

Conditions	5
Paths	8

Size

Total Lines	23
Code Lines	13

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

<?php
    public static function compute(array $actualLabels, array $predictedLabels, array $labels = []): array
    {
        $labels = count($labels) === 0 ? self::getUniqueLabels($actualLabels) : array_flip($labels);
        $matrix = self::generateMatrixWithZeros($labels);

It seems like $labels can also be of type null; however, parameter $labels of Phpml\Metric\ConfusionMatrix::generateMatrixWithZeros() does only seem to accept array, maybe add an additional type check?

        foreach ($actualLabels as $index => $actual) {
            $predicted = $predictedLabels[$index];

            if (!isset($labels[$actual], $labels[$predicted])) {
                continue;
            }

            if ($predicted === $actual) {
                $row = $column = $labels[$actual];
            } else {
                $row = $labels[$actual];
                $column = $labels[$predicted];
            }

            ++$matrix[$row][$column];
        }

        return $matrix;
    }