diff --git a/train.py b/train.py index f44634667..9ed85af59 100644 --- a/train.py +++ b/train.py @@ -12,7 +12,7 @@ y_test = np.genfromtxt("data/test_labels.csv") # Fit a model -depth = 5 +depth = 6 clf = RandomForestClassifier(max_depth=depth) clf.fit(X_train, y_train)