learn.get_preds() returns a tensor containing y_true like so:
[0 0 1 1 0 1].
How do I know whether
0 represents class
red if I had two classify on colors, i.e?
I think I found one solution here:
learn.dls.vocab returns a list of the class labels. My best guess is that the index refers to the class label, i.e. for
['cat', 'dog'] it would be
0 = cat, 1 = dog
Correct. The only case where this isn’t true is doing text, as vocab is the text vocab. There (though this is valid in any application too), you should do