classifierfactory bug fox

firefly-cpp · Dec 5, 2020 · 38cc26f · 38cc26f
1 parent de8b1bb
commit 38cc26f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/niaaml/classifiers/utility.py b/niaaml/classifiers/utility.py
@@ -31,7 +31,7 @@ def _set_parameters(self, **kwargs):
             'AdaBoost': classifiers.AdaBoost,
             'Bagging': classifiers.Bagging,
             'ExtremelyRandomizedTrees': classifiers.ExtremelyRandomizedTrees,
-            'LinearSVCClassifier': classifiers.LinearSVC,
+            'LinearSVC': classifiers.LinearSVC,
             'MultiLayerPerceptron': classifiers.MultiLayerPerceptron,
             'RandomForest': classifiers.RandomForest
         }