diff --git a/backend/core/generate_evaluation_plots.py b/backend/core/generate_evaluation_plots.py index caf0a80..bbc4150 100644 --- a/backend/core/generate_evaluation_plots.py +++ b/backend/core/generate_evaluation_plots.py @@ -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']