Selaa lähdekoodia

modify: 代码优化

yq 3 päivää sitten
vanhempi
sitoutus
19199f0552
1 muutettua tiedostoa jossa 6 lisäystä ja 1 poistoa
  1. 6 1
      feature/bin/strategy_norm.py

+ 6 - 1
feature/bin/strategy_norm.py

@@ -206,11 +206,16 @@ class StrategyNorm(FeatureStrategyBase):
         from IPython import display
 
         max_feature_num = self.ml_config.max_feature_num
+        filter_fast = context.get(ContextEnum.FILTER_FAST)
 
         f_display_title(display, "样本分布")
         display.display(metric_value_dict["样本分布"].table)
 
-        filter_fast = context.get(ContextEnum.FILTER_FAST)
+        df_importance = filter_fast["detail"]
+        df_importance = df_importance[df_importance["feature"].isin(self.x_columns)]
+        f_display_title(display, "入模变量")
+        display.display(df_importance)
+
         f_display_title(display, "快速筛选过程")
         print(f"剔除变量重要性排名{max_feature_num}以后的变量")
         print(filter_fast.get("overview"))