Update SVR.php

RubixML · Aug 27, 2024 · 146a0c8 · 146a0c8
1 parent fef1033
commit 146a0c8
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);
     }
 
     /**