Added code to map outputs back to the labels
#4
by nadaaaita - opened
I added code to extract the probabilities and map them back to the class labels. I was a bit confused about how to do it and it took me a minute to figure it out. So I thought this might be helpful to others.
I added code to extract the probabilities and map them back to the class labels. I was a bit confused about how to do it and it took me a minute to figure it out. So I thought this might be helpful to others.
Thank you !!!