search

vsubbian / WindowSHAP

A model-agnostic framework for explaining time-series classifiers using Shapley values
15 stars 8 forks source link

Plotting functions. #1

Closed franciscomalveiro closed 1 year ago

franciscomalveiro commented 1 year ago

Hello! First of all, thanks for the development of this tool, which I've been using it to generate explanations for the predictions of an LSTM model. By following the original paper, I've noticed there are some nice plots displayed, however in the pip package I cannot find the code that plots them.

By browsing through this repository and the provided notebook, I believe I've found functions on the util module which perform the plotting. For that reason, would it be possible to make these functions publicly available on the mentioned package?

Thanks!

aminnayebi commented 1 year ago

Hi,

Thank you for your interest in our method. I am sorry for a late reply. I am working on making the plots available as a method of the explanation classes so that it would be easier to use them. However, until then, you can use the heatmap() function in the util.py file to plot the time series alongside their explanations.

franciscomalveiro commented 1 year ago

Will do then, thanks!