CitrineInformatics / lolo

A random forest
Apache License 2.0
41 stars 12 forks source link

Pne 424 multitask classification shapley #316

Closed mVenetos97 closed 5 months ago

mVenetos97 commented 5 months ago

Extends Shapley to MultiTaskTree and ClassificationTree models. For the MultiTaskTree models Shapley Values are returned for each individual task label (e.g. if model predicts y_1 and y_2, we obtain separate Shapley values for y_1 and y_2). ClassificationTree models follow the same implementation as regression trees but are based on numeric class labels.