فهرست منبع

modify: 代码优化

yq 1 هفته پیش
والد
کامیت
12faa0a064
1فایلهای تغییر یافته به همراه1 افزوده شده و 2 حذف شده
  1. 1 2
      feature/woe/strategy_woe.py

+ 1 - 2
feature/woe/strategy_woe.py

@@ -296,7 +296,6 @@ class StrategyWoe(FeatureStrategyBase):
 
         train_data = data.train_data
         test_data = data.test_data
-        data = data.data
 
         if len(x_columns) == 0:
             x_columns = train_data.columns.tolist()
@@ -305,7 +304,7 @@ class StrategyWoe(FeatureStrategyBase):
         for column in columns_exclude:
             if column in x_columns:
                 x_columns.remove(column)
-        check_msg = DataExplore.check_type(data[x_columns])
+        check_msg = DataExplore.check_type(data.data[x_columns])
         if check_msg != "":
             print(f"数据类型分析:\n{check_msg}\n同一变量请保持数据类型一致")
             raise GeneralException(ResultCodesEnum.ILLEGAL_PARAMS, message=f"数据类型错误.")