FIX: remove early_stopping_rounds
Some checks failed
Build docker container / Build image (push) Failing after 6m56s
Some checks failed
Build docker container / Build image (push) Failing after 6m56s
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user