Update OneClassSVM.php

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