From 53a9ca600e707838f6e7fb92b65d0983d2a106b4 Mon Sep 17 00:00:00 2001 From: Elle O'Brien Date: Mon, 8 Jun 2020 14:43:22 -0700 Subject: [PATCH] increase forest depth --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index b7ae23a..e2bd9c8 100644 --- a/train.py +++ b/train.py @@ -13,7 +13,7 @@ # Fit a model -depth = 5 +depth = 7 clf = RandomForestClassifier(max_depth=depth) clf.fit(X_train,y_train)