From c0e30748a8e99156037b2ac0121e8bce5657a03d Mon Sep 17 00:00:00 2001
From: plagosh <plagosh@gmail.com>
Date: Tue, 14 Nov 2023 12:45:34 -0300
Subject: [PATCH] modificando forest depth

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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