Getting the native model for compatibility with shap.TreeExplainer

erdogant / hgboost

hgboost is a python package for hyper-parameter optimization for xgboost, catboost or lightboost using cross-validation, and evaluating the results on an independent validation set. hgboost can be applied for classification and regression tasks.

http://erdogant.github.io/hgboost

Other

57 stars 17 forks source link

Getting the native model for compatibility with shap.TreeExplainer #10

Closed nicolasaldecoa closed 2 years ago

nicolasaldecoa commented 2 years ago

Hello, first of all really nice project. I've just found out about it today and started playing with it a little bit. Is there any way to get the trained model as an XGBoost, LightGBM or CatBoost class in order to fit a shap.TreeExplainer instance to it?

Thanks in advance! -Nicolás

nicolasaldecoa commented 2 years ago

Nevermind! I checked the code and it is stored under the attribute "model".

erdogant commented 2 years ago

great!