From f3be2d0e232d297667b025da0aa59c403840154a Mon Sep 17 00:00:00 2001 From: dfcarvajal Date: Tue, 6 Jan 2026 23:47:12 +0100 Subject: [PATCH] FIX: remove early_stopping_rounds --- src/models_train.py | 28 ++++------------------------ 1 file changed, 4 insertions(+), 24 deletions(-) diff --git a/src/models_train.py b/src/models_train.py index da07a78..186f1e0 100644 --- a/src/models_train.py +++ b/src/models_train.py @@ -32,12 +32,7 @@ def train_demand_model(df): n_jobs=-1 ) - model.fit( - X_train, y_train, - eval_set=[(X_val, y_val)], - early_stopping_rounds=50, - verbose=False - ) + model.fit(X_train, y_train, verbose=False) return model @@ -79,12 +74,7 @@ def train_nulo_model(df): n_jobs=-1 ) - model.fit( - X_train, y_train, - eval_set=[(X_val, y_val)], - early_stopping_rounds=30, - verbose=False - ) + model.fit(X_train, y_train, verbose=False) return model @@ -199,12 +189,7 @@ def train_demand_model_advanced(df): n_jobs=-1 ) - model.fit( - X_train, y_train, - eval_set=[(X_train, y_train), (X_val, y_val)], - early_stopping_rounds=100, - verbose=False - ) + model.fit(X_train, y_train, verbose=False) return model @@ -260,11 +245,6 @@ def train_nulo_model_advanced(df): n_jobs=-1 ) - model.fit( - X_train, y_train, - eval_set=[(X_train, y_train), (X_val, y_val)], - early_stopping_rounds=50, - verbose=False - ) + model.fit(X_train, y_train, verbose=False) return model \ No newline at end of file