xgboost pipeline

记录：

 

from sklearn2pmml.pipeline import PMMLPipeline

other_params = {'learning_rate': 0.1,
    'seed': 0,
    'gamma': 0.05,
    'n_estimators': 15,
    'min_child_weight': 1.4,
    'colsample_bytree': 0.7,
    'max_depth': 2,
    'subsample': 0.8,
    'reg_alpha': 0.98,
    'reg_lambda':0.95}


pipeline_obj = PMMLPipeline([
    ('Xgbc', xgboost.XGBClassifier(**other_params))
])

pipeline_obj.fit(train_df[features],train_df[target])