Only models with p_value < 0.05

MOZI-AI · Mar 18, 2019 · 3d213a8 · 3d213a8
1 parent c830b01
commit 3d213a8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/crossval/moses_cross_val.py b/crossval/moses_cross_val.py
@@ -181,8 +181,9 @@ def count_features(self, models):
         """
 
         for moses_model in models:
-            tree = combo_parser.parse(moses_model.model)
-            self.tree_transformer.transform(tree)
+            if moses_model.test_score.p_value < 0.05:  # include only significant models in the feature count
+                tree = combo_parser.parse(moses_model.model)
+                self.tree_transformer.transform(tree)
 
     def majority_vote(self):
         """