Predicting On New Text With Fine-Tuned Multi-Label Model

Beginners
4

To get all scores, pipeline has a parameter

clf("text", return_all_scores=True)

For the label being LABEL_7, you need to check out the config.json in your repo. See for example id2label and label2id in config.json · distilbert-base-uncased-finetuned-sst-2-english at main.

5 Likes