Update train.py

Abhishek-N-D · Jul 24, 2020 · 6e0ee0a · 6e0ee0a
1 parent 926b9b4
commit 6e0ee0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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=2, random_state=seed)
 regr.fit(X_train, y_train)
 
 # Report training set score