From b4871b723f4883ac07085fed8b326e03fcdf21ee Mon Sep 17 00:00:00 2001 From: Abhishek-N-D <112235635+Abhishek-N-D@users.noreply.github.com> Date: Tue, 30 Jan 2024 21:50:27 +0530 Subject: [PATCH] Update train.py twice --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 236750b1..a20da118 100644 --- a/train.py +++ b/train.py @@ -13,7 +13,7 @@ # Fit a model -depth = 5 +depth = 3 clf = RandomForestClassifier(max_depth=depth) clf.fit(X_train, y_train)