fix fuzzy const settings (#51)

janoPig · Jan 21, 2024 · 0bff9b6 · 0bff9b6
1 parent 394f110
commit 0bff9b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/HROCH/fuzzy.py b/HROCH/fuzzy.py
@@ -127,8 +127,8 @@ class FuzzyRegressor(SymbolicSolver, ClassifierMixin):
         - opt_metric : (make_scorer) Scoring method
     """
 
-    FUZZY_INIT_CONST_SETTINGS = {'const_min':0.0, 'const_max':1.0, 'predefined_const_prob':0.0, 'predefined_const_set': []}
-    FUZZY_CONST_SETTINGS = {'const_min':0.0, 'const_max':1.0, 'predefined_const_prob':0.0, 'predefined_const_set': []}
+    INIT_CONST_SETTINGS = {'const_min':0.0, 'const_max':1.0, 'predefined_const_prob':0.0, 'predefined_const_set': []}
+    CONST_SETTINGS = {'const_min':0.0, 'const_max':1.0, 'predefined_const_prob':0.0, 'predefined_const_set': []}
 
     def __init__(self,
                  num_threads: int = 1,

diff --git a/test/ensemble_test.py b/test/ensemble_test.py
@@ -48,7 +48,7 @@ def test_voting_regressor(self):
 
     def test_bagging_classifier(self):
         for c in [NonlinearLogisticRegressor, FuzzyRegressor]:
-            ensemble_model = BaggingClassifier(estimator=NonlinearLogisticRegressor(**self.params))
+            ensemble_model = BaggingClassifier(estimator=c(**self.params))
             ensemble_model.fit(self.X_train, self.y_train)
             y_pred = ensemble_model.predict(self.X_test)
             self.assertEqual(y_pred.shape[0], self.y_test.shape[0])