mirror of
https://github.com/Llewellynvdm/php-ml.git
synced 2024-09-27 22:49:04 +00:00
parent
136a92c82b
commit
3e2708de17
@ -138,9 +138,11 @@ class DecisionTreeLeaf
|
|||||||
} else {
|
} else {
|
||||||
$col = "col_$this->columnIndex";
|
$col = "col_$this->columnIndex";
|
||||||
}
|
}
|
||||||
if (!preg_match('/^[<>=]{1,2}/', $value)) {
|
|
||||||
|
if (!preg_match('/^[<>=]{1,2}/', strval($value))) {
|
||||||
$value = "=$value";
|
$value = "=$value";
|
||||||
}
|
}
|
||||||
|
|
||||||
$value = "<b>$col $value</b><br>Gini: ".number_format($this->giniIndex, 2);
|
$value = "<b>$col $value</b><br>Gini: ".number_format($this->giniIndex, 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -0,0 +1,26 @@
|
|||||||
|
<?php
|
||||||
|
|
||||||
|
declare(strict_types=1);
|
||||||
|
|
||||||
|
namespace tests\Classification\DecisionTree;
|
||||||
|
|
||||||
|
use Phpml\Classification\DecisionTree\DecisionTreeLeaf;
|
||||||
|
use PHPUnit\Framework\TestCase;
|
||||||
|
|
||||||
|
class DecisionTreeLeafTest extends TestCase
|
||||||
|
{
|
||||||
|
public function testHTMLOutput()
|
||||||
|
{
|
||||||
|
$leaf = new DecisionTreeLeaf();
|
||||||
|
$leaf->value = 1;
|
||||||
|
$leaf->columnIndex = 0;
|
||||||
|
|
||||||
|
$rightLeaf = new DecisionTreeLeaf();
|
||||||
|
$rightLeaf->value = '<= 2';
|
||||||
|
$rightLeaf->columnIndex = 1;
|
||||||
|
|
||||||
|
$leaf->rightLeaf = $rightLeaf;
|
||||||
|
|
||||||
|
$this->assertEquals('<table ><tr><td colspan=3 align=center style=\'border:1px solid;\'><b>col_0 =1</b><br>Gini: 0.00</td></tr><tr><td></td><td> </td><td valign=top align=right><b>No |</b><br><table ><tr><td colspan=3 align=center style=\'border:1px solid;\'><b>col_1 <= 2</b><br>Gini: 0.00</td></tr></table></td></tr></table>', $leaf->getHTML());
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user