when do_lcs_sweep is false set rule compaction to None

UrbsLab · Aug 13, 2024 · 9ad5ace · 9ad5ace
1 parent f89c9d6
commit 9ad5ace
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/streamline/models/learning_based.py b/streamline/models/learning_based.py
@@ -88,6 +88,9 @@ def __init__(self, cv_folds=3, scoring_metric='balanced_accuracy',
             self.param_grid['N'] = [N, ]
         if nu:
             self.param_grid['nu'] = [nu, ]
+        if len(self.param_grid['learning_iterations']) == 1 and len(self.param_grid['N']) == 1 and \
+            len(self.param_grid['nu']) == 1:
+            self.param_grid['rule_compaction'] = [None, ]
         self.param_grid['expert_knowledge'] = expert_knowledge
         self.param_grid['random_state'] = [random_state, ]
         self.small_name = "ExSTraCS"