diff --git a/src/gui/panels/step8_ml_train_panel.py b/src/gui/panels/step8_ml_train_panel.py index 9a7f7ba..eaf17cb 100644 --- a/src/gui/panels/step8_ml_train_panel.py +++ b/src/gui/panels/step8_ml_train_panel.py @@ -39,6 +39,8 @@ PREPROC_CHINESE = { 'D2': '二阶导数 (D2)', 'DT': '去趋势 (DT)', 'CT': '中心化 (CT)', + 'WVAE': '小波去噪 (WVAE)', + 'DualStream_MNF': '双流MNF (物理指数+降维)', } # 模型类型:内部键 -> 显示文本 @@ -175,7 +177,7 @@ class Step8MlTrainPanel(QWidget): preproc_grid = QGridLayout() self.preproc_checkboxes = {} - preproc_methods = ['None', 'MMS', 'SS', 'SNV', 'MA', 'SG', 'MSC', 'D1', 'D2', 'DT', 'CT'] + preproc_methods = ['None', 'MMS', 'SS', 'SNV', 'MA', 'SG', 'MSC', 'D1', 'D2', 'DT', 'CT', 'WVAE', 'DualStream_MNF'] for i, method in enumerate(preproc_methods): checkbox = QCheckBox(PREPROC_CHINESE.get(method, method)) diff --git a/src/new/views/step8_view.py b/src/new/views/step8_view.py index 3314d22..cbab4d5 100644 --- a/src/new/views/step8_view.py +++ b/src/new/views/step8_view.py @@ -41,6 +41,8 @@ PREPROC_CHINESE = { "D2": "二阶导数 (D2)", "DT": "去趋势 (DT)", "CT": "中心化 (CT)", + "WVAE": "小波去噪 (WVAE)", + "DualStream_MNF": "双流MNF (物理指数+降维)", } MODEL_CHINESE = { @@ -68,7 +70,7 @@ SPLIT_CHINESE = { "random": "随机划分 (Random)", } -PREPROC_METHODS = ["None", "MMS", "SS", "SNV", "MA", "SG", "MSC", "D1", "D2", "DT", "CT"] +PREPROC_METHODS = ["None", "MMS", "SS", "SNV", "MA", "SG", "MSC", "D1", "D2", "DT", "CT", "WVAE", "DualStream_MNF"] MODEL_GROUPS = [ ("【线性模型】", ["LinearRegression", "Ridge", "Lasso", "ElasticNet", "PLS"]), ("【树模型】", ["DecisionTree", "RF", "ExtraTrees", "XGBoost", "LightGBM", "CatBoost"]),