Merge pull request #780 from Marchlak/master

Fix ix tests/tests_algorithms/test_nn.py::MultiClassNeuralNetworkAlgorithmTest::test_fit_predict
mljar · Sep 27, 2024 · 48927c5 · 48927c5
2 parents e5fd66e + ccd78e1
commit 48927c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tests_algorithms/test_nn.py b/tests/tests_algorithms/test_nn.py
@@ -138,7 +138,7 @@ def setUpClass(cls):
             "num_class": 3,
         }
 
-        lb = preprocessing.LabelBinarizer()
+        lb = preprocessing.LabelEncoder()
         lb.fit(cls.y)
         cls.y = lb.transform(cls.y)