search

bmurauer / pipelinehelper

scikit-helper to hot-swap pipeline elements
GNU General Public License v3.0
21 stars 9 forks source link

XGBoost support? #9

Closed browshanravan closed 4 years ago

browshanravan commented 4 years ago

I was wondering if you have any future plans to provide PipelineHelper support for XGBoost classifier since it has a great scikit-learn API and seems to perform in my top 3.

Many thanks in advance.

browshanravan commented 4 years ago

I just tried XGBoost with pipelinehelper using the latest scikit-learn and it works perfectly fine. I used to get a strange KeyError: 'base_score' but it turned out to be an issue from sklearn, which is resolved by the update. pipelinehelper is a fantastic package! thanks you!