search

TeamHG-Memex / eli5

A library for debugging/inspecting machine learning classifiers and explaining their predictions
http://eli5.readthedocs.io
MIT License
2.76k stars 334 forks source link

eli5.show_prediction doesn't work for my custom trained model #420

Open KiaDavari opened 2 years ago

KiaDavari commented 2 years ago

I have trained a binary classification model with Keras on a custom dataset. Want to check what is in the image that made the model predict a certain class over another? It worked for only one image and I am no longer able to test it on the other images. Does it mean my model is corrupted? Or is it a bug of ELI5? These are the versions of my packages in the notebook:

tensorflow==1.15.2
scikit-learn==0.21.3
keras==2.3.1
eli5==0.10.1

Thanks in advance