PAIR-code / what-if-tool

Source code/webpage/demos for the What-If Tool
https://pair-code.github.io/what-if-tool
Apache License 2.0
921 stars 170 forks source link

Does what-if-tool support lightgbm or xgboost or scikit learning models? #207

Closed wangbingnan136 closed 2 years ago

jameswex commented 2 years ago

Yes, it can support any model framework through use of the "custom prediction function" feature where you provide a python function that performs the model prediction. That function can make use of xgboost, scikit, or any other model framework. See https://pair-code.github.io/what-if-tool/learn/tutorials/custom-prediction/ for more details.