aisi-tech/forsetsystem
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: backend/core/generate_evaluation_plots.py

Browse Source
This commit is contained in:
shenjianZ
2026-03-20 17:05:02 +08:00
parent 6d42d9dac3
commit eab1a62ffb
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/core/generate_evaluation_plots.py
View File

@@ -26,7 +26,6 @@ from core.model_features import (
fit_outlier_bounds,
make_target_bins,
normalize_columns,
prepare_modeling_dataframe,
)
from core.preprocessing import get_clean_data
@@ -95,11 +94,9 @@ def load_lstm_predictions():
raise RuntimeError('无法加载深度学习模型，请确认 torch 环境和模型文件正常。')
raw_train_df, raw_test_df = get_test_split()
fit_df = prepare_modeling_dataframe(raw_train_df)
test_df = prepare_modeling_dataframe(raw_test_df)
outlier_bounds = fit_outlier_bounds(fit_df, NUMERICAL_OUTLIER_COLUMNS)
fit_df = apply_outlier_bounds(fit_df, outlier_bounds)
test_df = apply_outlier_bounds(test_df, outlier_bounds)
outlier_bounds = fit_outlier_bounds(raw_train_df, NUMERICAL_OUTLIER_COLUMNS)
fit_df = apply_outlier_bounds(raw_train_df, outlier_bounds)
test_df = apply_outlier_bounds(raw_test_df, outlier_bounds)
feature_layout = bundle['feature_layout']
category_maps = bundle['category_maps']