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.

Healthcare and Clinical NLP org

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 !!!

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment