raghakot / keras-vis

Neural network visualization toolkit for keras
https://raghakot.github.io/keras-vis
MIT License
2.98k stars 660 forks source link

Doing saliency maps for a regression problem #70

Open amunozj opened 7 years ago

amunozj commented 7 years ago

Hi! first of all I wanted to thank you for putting this together it's pretty awesome. Is there a quick hack I could do use your saliency visualizations for a regression problem. That is, instead of providing and index you provide a value.

I tried doing something like this:

heatmap = visualize_saliency(model, layer_idx, filter_indices=0.00054109, seed_input=dfnp)

But I get an error:

Value passed to parameter 'begin' has DataType float32 not in list of allowed values: int32, int64

thank you!

raghakot commented 7 years ago

See the self-driving car example in the applications folder. It shows how to handle regression use-cases