|
|
@@ -36,6 +36,7 @@ class StrategyNorm(FeatureStrategyBase):
|
|
|
format_bin = self.ml_config.format_bin
|
|
|
params_xgb = self.ml_config.params_xgb
|
|
|
max_feature_num = self.ml_config.max_feature_num
|
|
|
+ columns_anns = self.ml_config.columns_anns
|
|
|
|
|
|
train_data = data.train_data.copy()
|
|
|
test_data = data.test_data.copy()
|
|
|
@@ -118,7 +119,8 @@ class StrategyNorm(FeatureStrategyBase):
|
|
|
feature.append(x_column)
|
|
|
importance_weight.append(feature_cache)
|
|
|
|
|
|
- df_importance = pd.DataFrame({'feature': feature, f'importance_weight': importance_weight})
|
|
|
+ anns = [columns_anns.get(column, "-") for column in feature]
|
|
|
+ df_importance = pd.DataFrame({'feature': feature, f'importance_weight': importance_weight, "释义": anns})
|
|
|
df_importance.sort_values(by=["importance_weight"], ascending=[False], inplace=True)
|
|
|
df_importance.reset_index(drop=True, inplace=True)
|
|
|
df_importance_rank = df_importance[df_importance["importance_weight"] > 0]
|
