From d425a05740a7c403485107b2c97ad2b104088934 Mon Sep 17 00:00:00 2001
From: ecm9999 <pcunabefrain@gmail.com>
Date: Sat, 15 Jun 2024 00:18:52 -0400
Subject: [PATCH] Update train.py

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

diff --git a/train.py b/train.py
index 25a64304f..75c002b5d 100644
--- a/train.py
+++ b/train.py
@@ -23,7 +23,7 @@
 #################################
 
 # Fit a model on the train section
-regr = RandomForestRegressor(max_depth=2, random_state=seed)
+regr = RandomForestRegressor(max_depth=20, random_state=seed)
 regr.fit(X_train, y_train)
 
 # Report training set score