From df9bb8cbb8dee523f614d44ef33f1f6f262fce2d Mon Sep 17 00:00:00 2001 From: 87owo <85057800+87owo@users.noreply.github.com> Date: Thu, 7 May 2026 14:12:56 +0800 Subject: [PATCH] Add files via upload --- Engine/Properties/train_server.py | 6 +++--- Engine/Properties/train_user.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/Engine/Properties/train_server.py b/Engine/Properties/train_server.py index b38c75c..ffdbeb3 100644 --- a/Engine/Properties/train_server.py +++ b/Engine/Properties/train_server.py @@ -14,14 +14,14 @@ DB_PATH = "pe_features.db" MODEL_FILE = "model.txt" ONNX_FILE = "Pefile_General_S1.onnx" FEATURE_FILE = "features.json" -TEST_SIZE = 0.1 +TEST_SIZE = 0.0001 RANDOM_SEED = 42 LGBM_PARAMS = { 'objective': 'binary', 'metric': ['binary_logloss', 'auc'], 'boosting_type': 'gbdt', - 'num_leaves': 512, + 'num_leaves': 256, 'learning_rate': 0.05, 'feature_fraction': 0.8, 'bagging_fraction': 0.8, @@ -30,7 +30,7 @@ LGBM_PARAMS = { 'seed': RANDOM_SEED, 'n_jobs': -1, 'max_depth': -1, - 'min_data_in_leaf': 10, + 'min_data_in_leaf': 20, 'lambda_l1': 0.05, 'lambda_l2': 0.05, } diff --git a/Engine/Properties/train_user.py b/Engine/Properties/train_user.py index c95de05..d3fad91 100644 --- a/Engine/Properties/train_user.py +++ b/Engine/Properties/train_user.py @@ -14,7 +14,7 @@ DB_PATH = "pe_features.db" MODEL_FILE = "model.txt" ONNX_FILE = "Pefile_General_T1.onnx" FEATURE_FILE = "features.json" -TEST_SIZE = 0.1 +TEST_SIZE = 0.0001 RANDOM_SEED = 42 LGBM_PARAMS = { @@ -30,7 +30,7 @@ LGBM_PARAMS = { 'seed': RANDOM_SEED, 'n_jobs': -1, 'max_depth': -1, - 'min_data_in_leaf': 10, + 'min_data_in_leaf': 20, 'lambda_l1': 0.05, 'lambda_l2': 0.05, }