Update StrategyLearner to pass tests

2020-11-04 17:32:02 -05:00
parent 10d87aefd3
commit 5fbbc26929
4 changed files with 22 additions and 58 deletions
--- a/strategy_evaluation/AbstractTreeLearner.py
+++ b/strategy_evaluation/AbstractTreeLearner.py
@@ -47,9 +47,9 @@ class AbstractTreeLearner:

        if xs.shape[0] <= self.leaf_size:
            value = np.mean(y)
-            if value < -0.3:
+            if value < -0.2:
                value = -1
-            elif value > 0.3:
+            elif value > 0.2:
                value = 1
            else:
                value = 0