From 9eb68d1d114379c1cfe68d6021235df0a1a4871e Mon Sep 17 00:00:00 2001 From: jfcalder Date: Tue, 31 Oct 2023 16:57:06 -0300 Subject: [PATCH] modificando profundidad forest --- 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)