mirror of
https://github.com/Karaka-Management/phpOMS.git
synced 2026-01-11 17:58:41 +00:00
89 lines
2.6 KiB
PHP
Executable File
89 lines
2.6 KiB
PHP
Executable File
<?php
|
|
/**
|
|
* Jingga
|
|
*
|
|
* PHP Version 8.2
|
|
*
|
|
* @package tests
|
|
* @copyright Dennis Eichhorn
|
|
* @license OMS License 2.0
|
|
* @version 1.0.0
|
|
* @link https://jingga.app
|
|
*/
|
|
declare(strict_types=1);
|
|
|
|
namespace phpOMS\tests\Math\Stochastic;
|
|
|
|
use phpOMS\Math\Stochastic\NaiveBayesClassifier;
|
|
|
|
/**
|
|
* @internal
|
|
*/
|
|
#[\PHPUnit\Framework\Attributes\TestDox('phpOMS\tests\Math\Stochastic\NaiveBayesClassifierTest: Naive bayes classifier for numeric values and strings/attributes')]
|
|
final class NaiveBayesClassifierTest extends \PHPUnit\Framework\TestCase
|
|
{
|
|
public const PLAY = [
|
|
['weather' => ['Overcast']],
|
|
['weather' => ['Rainy']],
|
|
['weather' => ['Sunny']],
|
|
['weather' => ['Sunny']],
|
|
['weather' => ['Overcast']],
|
|
['weather' => ['Sunny']],
|
|
['weather' => ['Rainy']],
|
|
['weather' => ['Overcast']],
|
|
['weather' => ['Overcast']],
|
|
];
|
|
|
|
public const NO_PLAY = [
|
|
['weather' => ['Sunny']],
|
|
['weather' => ['Rainy']],
|
|
['weather' => ['Rainy']],
|
|
['weather' => ['Sunny']],
|
|
['weather' => ['Rainy']],
|
|
];
|
|
|
|
public const MALE = [
|
|
['height' => 6, 'weight' => 180, 'foot' => 12],
|
|
['height' => 5.92, 'weight' => 190, 'foot' => 11],
|
|
['height' => 5.58, 'weight' => 170, 'foot' => 12],
|
|
['height' => 5.92, 'weight' => 165, 'foot' => 10],
|
|
];
|
|
|
|
public const FEMALE = [
|
|
['height' => 5, 'weight' => 100, 'foot' => 6],
|
|
['height' => 5.5, 'weight' => 150, 'foot' => 8],
|
|
['height' => 5.42, 'weight' => 130, 'foot' => 7],
|
|
['height' => 5.75, 'weight' => 150, 'foot' => 9],
|
|
];
|
|
|
|
#[\PHPUnit\Framework\Attributes\Group('framework')]
|
|
#[\PHPUnit\Framework\Attributes\TestDox('The classification of strings/attributes is correct')]
|
|
public function testTextClassifier() : void
|
|
{
|
|
$filter = new NaiveBayesClassifier();
|
|
$filter->train('play', self::PLAY);
|
|
$filter->train('noplay', self::NO_PLAY);
|
|
|
|
self::assertEqualsWithDelta(
|
|
0.6,
|
|
$filter->matchCriteria('play', ['weather' => ['Sunny']], 1),
|
|
0.01
|
|
);
|
|
}
|
|
|
|
#[\PHPUnit\Framework\Attributes\Group('framework')]
|
|
#[\PHPUnit\Framework\Attributes\TestDox('The classification of numeric values is correct')]
|
|
public function testNumericClassifier() : void
|
|
{
|
|
$filter = new NaiveBayesClassifier();
|
|
$filter->train('male', self::MALE);
|
|
$filter->train('female', self::FEMALE);
|
|
|
|
self::assertEqualsWithDelta(
|
|
0.999988,
|
|
$filter->matchCriteria('female', ['height' => 6, 'weight' => 130, 'foot' => 8]),
|
|
0.01
|
|
);
|
|
}
|
|
}
|