From d73edd039fbaf1d769d9bcd62b5f53cbfcc4b7ff Mon Sep 17 00:00:00 2001 From: Macedo Maia Date: Mon, 8 Apr 2024 10:02:37 +0200 Subject: [PATCH 1/2] changes bla --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 6e3d6906..b11b562f 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 = 6 clf = RandomForestClassifier(max_depth=depth) clf.fit(X_train, y_train) From e4b7270342bfc91b77162ff3b0c1e68e35b49394 Mon Sep 17 00:00:00 2001 From: Macedo Maia Date: Mon, 8 Apr 2024 10:15:12 +0200 Subject: [PATCH 2/2] changes bla --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index b11b562f..4063a8bd 100644 --- a/train.py +++ b/train.py @@ -10,7 +10,7 @@ y_test = np.genfromtxt("data/test_labels.csv") # Fit a model -depth = 6 +depth = 8 clf = RandomForestClassifier(max_depth=depth) clf.fit(X_train, y_train)