octoleo/php-ml
1
0
Fork 0
mirror of https://github.com/Llewellynvdm/php-ml.git synced 2024-11-16 18:25:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

svm is hard :(

Browse Source
This commit is contained in:
Arkadiusz Kondas 2016-04-21 23:21:08 +02:00
parent 37782eba98
commit 118ee0b5fe
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/Phpml/Classifier/SupportVectorMachine.php
View File

@ -6,20 +6,21 @@ namespace Phpml\Classifier;
use Phpml\Classifier\Traits\Predictable; use Phpml\Classifier\Traits\Predictable;
use Phpml\Classifier\Traits\Trainable; use Phpml\Classifier\Traits\Trainable;
use Phpml\Math\Kernel;
class SupportVectorMachine implements Classifier class SupportVectorMachine implements Classifier
{ {
use Trainable, Predictable; use Trainable, Predictable;
/** /**
* @var float * @var Kernel
*/ */
private $gamma; private $kernel;
/** /**
* @var float * @var float
*/ */
private $epsilon; private $C;
/** /**
* @var float * @var float
@ -32,20 +33,23 @@ class SupportVectorMachine implements Classifier
private $upperBound; private $upperBound;
/** /**
* @param float $gamma * @param Kernel $kernel
* @param float $epsilon * @param float $C
* @param float $tolerance * @param float $tolerance
* @param int $upperBound * @param int $upperBound
*/ */
public function __construct(float $gamma = .5, float $epsilon = .001, float $tolerance = .001, int $upperBound = 100) public function __construct(Kernel $kernel = null, float $C = 1.0, float $tolerance = .001, int $upperBound = 100)
{ {
$this->gamma = $gamma; if (null === $kernel) {
$this->epsilon = $epsilon; $kernel = new Kernel\RBF($gamma = .001);
}
$this->kernel = $kernel;
$this->C = $C;
$this->tolerance = $tolerance; $this->tolerance = $tolerance;
$this->upperBound = $upperBound; $this->upperBound = $upperBound;
} }
/** /**
* @param array $sample * @param array $sample
* *