Getting filenames and corresponding predictions

[muellerzr (Zachary Mueller)]

PR: New Widget! Class Confusion - With Google Colab Support!

Hey everyone! Have you ever wanted to further explore where your model was being confused? Look and take a peek at which specific images were being confused for x vs y class? Or look at the distributions of your mislabeled classes for your tabular models? ClassConfusion will do exactly this! For both images and tabular, we pass in our ClassificationInterpretation object, along with a list of classes, and we can specify if we want an ordered set of classes or just the two back to back. For insta…

That’s the pr. The code I showed is how I got the filenames associated with when two classes matched. You should be able to extrapolate it from there. Source code link should be there too for my github as well.