Update SVR.php (#344)

RubixML · Aug 30, 2024 · c1bede9 · c1bede9
1 parent 65251d9
commit c1bede9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/Regressors/SVR.php b/src/Regressors/SVR.php
@@ -235,8 +235,12 @@ public function predictSample(array $sample)
         if (!$this->model) {
             throw new RuntimeException('Estimator has not been trained.');
         }
-
-        return $this->model->predict($sample);
+        //As SVM needs to have the same keys and order between training samples and those to predict we need to put an offset to the keys
+        $sampleWithOffset = [];
+        foreach($sample as $key=>$value){
+            $sampleWithOffset[$key+1] = $value;
+        }
+        return $this->model->predict($sampleWithOffset);
     }
 
     /**