FIX: remove early_stopping_rounds
Some checks failed
Build docker container / Build image (push) Failing after 6m56s

This commit is contained in:
2026-01-06 23:47:12 +01:00
parent 61d91a8e94
commit f3be2d0e23

View File

@@ -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