From 2e2873ea4d0f1d920ae0dcecb4bc2ba92bdcbed3 Mon Sep 17 00:00:00 2001 From: sangseconds <82400654+sangseconds@users.noreply.github.com> Date: Wed, 22 Nov 2023 20:58:41 +0700 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 92678f346..43b3a7873 100644 --- a/train.py +++ b/train.py @@ -23,7 +23,7 @@ ################################# # Fit a model on the train section -regr = RandomForestRegressor(max_depth=5, random_state=seed) +regr = RandomForestRegressor(max_depth=40, random_state=seed) regr.fit(X_train, y_train) # Report training set score