diff --git a/Engine/Properties/train_server.py b/Engine/Properties/train_server.py index aa2331c..07960ae 100644 --- a/Engine/Properties/train_server.py +++ b/Engine/Properties/train_server.py @@ -114,7 +114,7 @@ def train_process(X, y): n_neg = (y_train == 0).sum() base_weight = n_neg / n_pos if n_pos > 0 else 1.0 - weight_ratio_target = 0.01 + weight_ratio_target = 0.005 final_pos_weight = base_weight * weight_ratio_target print(f"[*] Balance Report: Safe={n_neg}, Malware={n_pos}") diff --git a/Engine/Properties/train_user.py b/Engine/Properties/train_user.py index 9a46d44..91deb55 100644 --- a/Engine/Properties/train_user.py +++ b/Engine/Properties/train_user.py @@ -21,7 +21,7 @@ LGBM_PARAMS = { 'objective': 'binary', 'metric': ['binary_logloss', 'auc'], 'boosting_type': 'gbdt', - 'num_leaves': 63, + 'num_leaves': 64, 'learning_rate': 0.05, 'feature_fraction': 0.8, 'bagging_fraction': 0.8,