5k nodes

SmokinCaterpillar · Feb 20, 2018 · dd11a20 · dd11a20
1 parent 424b5ef
commit dd11a20
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/do_cross_val.py b/scripts/do_cross_val.py
@@ -29,7 +29,7 @@ def main():
 
     gc.collect()
 
-    regressor_kwargs = dict(n_estimators=256, max_leaf_nodes=10000,
+    regressor_kwargs = dict(n_estimators=256, max_leaf_nodes=5000,
                               max_features=0.2, n_jobs=-1, verbose=1,
                               random_state=42)
 

diff --git a/trufflepig/main.py b/trufflepig/main.py
@@ -75,7 +75,7 @@ def main():
     else:
         post_frame = None
 
-    regressor_kwargs = dict(n_estimators=256, max_leaf_nodes=10000,
+    regressor_kwargs = dict(n_estimators=256, max_leaf_nodes=5000,
                               max_features=0.2, n_jobs=-1, verbose=1,
                               random_state=42)