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.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment