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