Update OneClassSVM.php

RubixML · Oct 30, 2024 · 011b52f · 011b52f
1 parent c3897a3
commit 011b52f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/AnomalyDetectors/OneClassSVM.php b/src/AnomalyDetectors/OneClassSVM.php
@@ -185,8 +185,10 @@ public function train(Dataset $dataset) : void
 
         $data = [];
 
-        foreach ($dataset->samples() as $i => $sample) {
-            $data[] = array_merge([1], $sample);
+        foreach ($dataset->samples() as $sample) {
+            $sample_array = $sample;
+            array_unshift($sample_array, 1);
+            $data[] = $sample_array;
         }
 
         $this->model = $this->svm->train($data);