How to Determine the Corresponding Category for Each Predicted Class ID?
#1
by
oussama120
- opened
Great work ! I have a question regarding the classification results. While the current implementation outputs the probabilities for each predicted class, I need an easy way to determine the corresponding category for each predicted class ID. For instance, if the predicted class ID is 36, how can I find out which category this ID corresponds to? Thanks in advance .
Hello @oussama120 ,
Thank you for your kind words. I added a section in the model card to show how to load and use our model. You can find it here.
ahmedheakl
changed discussion status to
closed