Explorar o código

modify: 代码优化

yq hai 1 semana
pai
achega
12faa0a064
Modificáronse 1 ficheiros con 1 adicións e 2 borrados
  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
         train_data = data.train_data
         test_data = data.test_data
         test_data = data.test_data
-        data = data.data
 
 
         if len(x_columns) == 0:
         if len(x_columns) == 0:
             x_columns = train_data.columns.tolist()
             x_columns = train_data.columns.tolist()
@@ -305,7 +304,7 @@ class StrategyWoe(FeatureStrategyBase):
         for column in columns_exclude:
         for column in columns_exclude:
             if column in x_columns:
             if column in x_columns:
                 x_columns.remove(column)
                 x_columns.remove(column)
-        check_msg = DataExplore.check_type(data[x_columns])
+        check_msg = DataExplore.check_type(data.data[x_columns])
         if check_msg != "":
         if check_msg != "":
             print(f"数据类型分析:\n{check_msg}\n同一变量请保持数据类型一致")
             print(f"数据类型分析:\n{check_msg}\n同一变量请保持数据类型一致")
             raise GeneralException(ResultCodesEnum.ILLEGAL_PARAMS, message=f"数据类型错误.")
             raise GeneralException(ResultCodesEnum.ILLEGAL_PARAMS, message=f"数据类型错误.")