search

christophM / interpretable-ml-book

Book about interpretable machine learning
https://christophm.github.io/interpretable-ml-book/
Other
4.75k stars 1.06k forks source link

How to interpret the ALE plot? #221

Closed xiaochi-liu closed 3 years ago

xiaochi-liu commented 3 years ago

Hello @christophM,

Many thanks for this outstanding book.

I have a small question about how to interpret the ALE plot. In section 5.3.3, you wrote:

image

So, what does the "average prediction" exactly mean? When I write the result section of a journal article, I think it's better to provide this average prediction. So how to calculate it?

Your kind guidance would be much appreciated!

christophM commented 3 years ago

The average prediction is when you take all of your data, get the prediction for it and average it

xiaochi-liu commented 3 years ago

Thank you very much @christophM. Can I get this average prediction using iml?