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