I am working on building XGBoost PMML using SKLearn and SKLearn2PMML. I am having some numerical,somecategorical and datetime columns from which i am creating new feature inside the pipeline. When i am trying to train the model, it gets failed as the original categorical features also gets passed to the final classfier by default. Is there any way to restrict the features by specifying the feature names ?


1 Answer 1


After digging down too much and some help from sklearn2pmml creator, I managed to filter the final columns to be passed to the classifier.

Note : Here recorder is DataFrameMapper object.

1.Getting categorical column indexes.

cat_cols = [recorder.transformed_names_.index(c) for c in categoricalCols if c in recorder.transformed_names_]

2.Adding ColumnTransformer to filter those column with the help of their indexes.

pipeline = PMMLPipeline([
    ("mapper", recorder),
    ("select", ColumnTransformer([("drop", "drop", cat_cols)], remainder='passthrough')),
    ("classifier", xgb.XGBClassifier())

3.Fitting the Data to the pipeline.


4.Creating PMML file out of Pipeline.

out_file = "XGBoost.pmml"
sklearn2pmml(pipeline, out_file)

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.