Handle y_true as class targets for GradCAM

sicara / tf-explain

Interpretability Methods for tf.keras models with Tensorflow 2.x

https://tf-explain.readthedocs.io

MIT License

1.02k stars 112 forks source link

Handle y_true as class targets for GradCAM #143

Open RaphaelMeudec opened 4 years ago

RaphaelMeudec commented 4 years ago

This PR makes it possible for the user to pass y_true instead of just giving a class_index. It's helpful as users won't need to build their dataset with one class only, and will just pass the dataset they use for training

What's left to do:

[ ] Fix tests
[ ] Extend the use of y_true to all current methods
[ ] Impact documentation