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)