|
@@ -99,9 +99,12 @@ class ModelXgb(ModelBase):
|
|
# early_stopping_rounds=params_xgb.get("early_stopping_rounds"),
|
|
# early_stopping_rounds=params_xgb.get("early_stopping_rounds"),
|
|
# verbose=params_xgb.get("verbose_eval"),
|
|
# 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]
|
|
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:
|
|
for column in x_columns_selected:
|
|
if column in str_columns_selected:
|
|
if column in str_columns_selected:
|
|
continue
|
|
continue
|