diff --git a/Engine/Properties/train_experimental.py b/Engine/Properties/train_experimental.py index b166e2b..041a44a 100644 --- a/Engine/Properties/train_experimental.py +++ b/Engine/Properties/train_experimental.py @@ -210,7 +210,7 @@ def train_process(X, y, feature_names): params, train_data, valid_sets=[valid_data], - num_boost_round=500, + num_boost_round=450, callbacks=[ lgb.log_evaluation(period=50), lgb.reset_parameter(learning_rate=_calculate_dynamic_lr) diff --git a/Engine/Properties/train_server.py b/Engine/Properties/train_server.py index f380c88..d36b56f 100644 --- a/Engine/Properties/train_server.py +++ b/Engine/Properties/train_server.py @@ -210,7 +210,7 @@ def train_process(X, y, feature_names): params, train_data, valid_sets=[valid_data], - num_boost_round=350, + num_boost_round=400, callbacks=[ lgb.log_evaluation(period=50), lgb.reset_parameter(learning_rate=_calculate_dynamic_lr)