Просмотр исходного кода

bugfix: 没有字符型变量时mapper错误

yq 1 день назад
Родитель
Сommit
176eadcfcf
1 измененных файлов с 4 добавлено и 1 удалено
  1. 4 1
      model/model_xgb.py

+ 4 - 1
model/model_xgb.py

@@ -99,9 +99,12 @@ class ModelXgb(ModelBase):
         #                early_stopping_rounds=params_xgb.get("early_stopping_rounds"),
         #                verbose=params_xgb.get("verbose_eval"),
         #                )
+        # self.model.save_model("./cache/train/demo/xgb_model1.bin")
 
         str_columns_selected = [i for i in x_columns_selected if i not in num_columns]
-        mapper = [(str_columns_selected, OneHotEncoder())]
+        mapper = []
+        if len(str_columns_selected) > 0:
+            mapper = [(str_columns_selected, OneHotEncoder())]
         for column in x_columns_selected:
             if column in str_columns_selected:
                 continue