search

interpretml / interpret

Fit interpretable models. Explain blackbox machine learning.
https://interpret.ml/docs
MIT License
6.24k stars 725 forks source link

Does EMB works for multi-label classification? #104

Closed hpsong88 closed 4 years ago

hpsong88 commented 4 years ago

Hello,

I am wondering if EBM works for multi-label classification or fits to OneVsRestClassifier?

Thanks!

interpret-ml commented 4 years ago

Hi @hpsong88,

EBM currently supports multiclass classification, but not multi-label classification. For multiclass, EBM doesn't use the OneVsRest method, or any multiclass -> binary reduction techniques. We use boosted decision trees under the hood to train EBM, which naturally handle multiclass learning.

-InterpretML Team