@yuxi_liu_1995 That is correct. My widget will be able to show you what images were most wrong in a particular category (box on the confusion matrix) which is about as close as I’ve gotten so far, along with the file name . See here:
https://forums.fast.ai/t/new-colab-widget-classlosses/50086/6