DAMO-NLP-SG
/

zero-shot-classify-SSTuning-base

Zero-Shot Classification

text-classification

Zero-Shot Classification

Model card Files Files and versions Community

lukecq commited on May 19, 2023

Commit

7599afa

•

1 Parent(s): defe81c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -35,9 +35,9 @@ tokenizer = AutoTokenizer.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuni
 model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-base")
 text = "I love this place! The food is always so fresh and delicious."
 list_label = ["negative","positve"]
 def add_prefix(text, list_label, label_num = 20, shuffle = False):
     list_label = [x+'.' if x[-1] != '.' else x for x in list_label]
     list_label_new = list_label + [tokenizer.pad_token]* (label_num - len(list_label))

 model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-base")
 text = "I love this place! The food is always so fresh and delicious."
 list_label = ["negative","positve"]
+list_ABC = [x for x in string.ascii_uppercase]
 def add_prefix(text, list_label, label_num = 20, shuffle = False):
     list_label = [x+'.' if x[-1] != '.' else x for x in list_label]
     list_label_new = list_label + [tokenizer.pad_token]* (label_num - len(list_label))