From 118ee0b5fed0f7d004b5da48182f6636e2678113 Mon Sep 17 00:00:00 2001
From: Arkadiusz Kondas <arkadiusz.kondas@gmail.com>
Date: Thu, 21 Apr 2016 23:21:08 +0200
Subject: [PATCH] svm is hard :(

---
 src/Phpml/Classifier/SupportVectorMachine.php | 22 +++++++++++--------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/src/Phpml/Classifier/SupportVectorMachine.php b/src/Phpml/Classifier/SupportVectorMachine.php
index 25d266c..e16ebbe 100644
--- a/src/Phpml/Classifier/SupportVectorMachine.php
+++ b/src/Phpml/Classifier/SupportVectorMachine.php
@@ -6,20 +6,21 @@ namespace Phpml\Classifier;
 
 use Phpml\Classifier\Traits\Predictable;
 use Phpml\Classifier\Traits\Trainable;
+use Phpml\Math\Kernel;
 
 class SupportVectorMachine implements Classifier
 {
     use Trainable, Predictable;
 
     /**
-     * @var float
+     * @var Kernel
      */
-    private $gamma;
+    private $kernel;
 
     /**
      * @var float
      */
-    private $epsilon;
+    private $C;
 
     /**
      * @var float
@@ -32,20 +33,23 @@ class SupportVectorMachine implements Classifier
     private $upperBound;
 
     /**
-     * @param float $gamma
-     * @param float $epsilon
+     * @param Kernel $kernel
+     * @param float $C
      * @param float $tolerance
      * @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;
-        $this->epsilon = $epsilon;
+        if (null === $kernel) {
+            $kernel = new Kernel\RBF($gamma = .001);
+        }
+
+        $this->kernel = $kernel;
+        $this->C = $C;
         $this->tolerance = $tolerance;
         $this->upperBound = $upperBound;
     }
 
-
     /**
      * @param array $sample
      *