From 9018c839681305a58a38e84043b1ef9a9f897df4 Mon Sep 17 00:00:00 2001 From: PauloQJ <163788184+PauloQJ@users.noreply.github.com> Date: Mon, 20 May 2024 22:56:00 -0400 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)