search

easeml / datascope

Measuring data importance over ML pipelines using the Shapley value.
https://ease.ml/datascope
MIT License
36 stars 4 forks source link

Is regression supported? #6

Open DSoudis opened 2 years ago

DSoudis commented 2 years ago

Looking at the code my understanding is that only classification is possible. (SklearnModelAccuracy) I tried using SklearnModelUtility with MSE as a metric but this leads to a NotImplementedError.

Thanks!

xzyaoi commented 2 years ago

Hi @DSoudis,

Yes there is only SklearnModelAccuracy implemented in this library for now (my understanding too). Let us think about if we can support regressions tasks and I will write back on this topic :)

Best regards, Xiaozhe