eli5
eli5 copied to clipboard
Occasional no highlight in show_prediction for images
I use ktrain wrapper library for Keras which calls
eli5.show_prediction(model, x)
[eli5: '0.10.1']
and for some images, there is no indication (color highlight) on the predicted image (both correct and incorrect classification). Ex:
HI @abedkhooli,
Could you post the result of eli5.explain_prediction(model, x)
? This should show if it's a problem with the explanation or with the color highlight.
You can also try changing the color highlight like this: eli5.show_prediction(model, x, alpha_limit=1.0, colormap=matplotlib.cm.cividis)
Ktrain uses a forked version with tf.keras and TF2.x, so these won't work. I tried with one less epoch and got it highlighted correctly!