Change some values

iterative-test · Jan 6, 2025 · 1b1ed18 · 1b1ed18
1 parent d049cf9
commit 1b1ed18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)