Merge pull request neurodata#323 from neurodata/EYezerets-patch-line237

forest.py line 237 change max_depth to self.max_depth
david-z-shi · Oct 18, 2020 · 8f20cb3 · 8f20cb3
2 parents f663fef + 23231f9
commit 8f20cb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/proglearn/forest.py b/proglearn/forest.py
@@ -258,7 +258,7 @@ def fit(self, X, y):
         self.lf_ = LifelongClassificationForest(
             n_estimators=self.n_estimators,
             default_finite_sample_correction=self.finite_sample_correction,
-            default_max_depth=max_depth,
+            default_max_depth=self.max_depth,
         )
         self.lf_.add_task(X, y, task_id=0)
         return self