Skip to content

Commit

Permalink
(lib docs) Mention heatmap is part of TargetExplanation
Browse files Browse the repository at this point in the history
  • Loading branch information
teabolt committed Jul 8, 2019
1 parent 5f05744 commit 20cb6ec
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/source/libraries/keras.rst
Original file line number Diff line number Diff line change
Expand Up @@ -19,11 +19,11 @@ explain_prediction
Currently ELI5 supports :func:`eli5.explain_prediction` for Keras image classifiers.
:func:`eli5.explain_prediction` explains image classifications through `Grad-CAM <https://arxiv.org/pdf/1610.02391.pdf>`_.

The returned :class:`eli5.base.Explanation` object has some important properties:
The returned :class:`eli5.base.Explanation` object contains some important objects:

* ``image`` represents the image that is inputted into the model.

* ``heatmap`` is a grayscale "localization map", roughly indicating regions of importance in the image for the predicted class.
* ``heatmap`` (property of each :class:`eli5.base.TargetExplanation` in ``targets``) is a grayscale "localization map", roughly indicating regions of importance in the image for the predicted class.


Important arguments to :func:`eli5.explain_prediction` for ``Model`` and ``Sequential``:
Expand Down

0 comments on commit 20cb6ec

Please sign in to comment.