Update README.md
Browse files
README.md
CHANGED
|
@@ -35,9 +35,9 @@ tokenizer = AutoTokenizer.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuni
|
|
| 35 |
model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-base")
|
| 36 |
|
| 37 |
text = "I love this place! The food is always so fresh and delicious."
|
| 38 |
-
|
| 39 |
list_label = ["negative","positve"]
|
| 40 |
|
|
|
|
| 41 |
def add_prefix(text, list_label, label_num = 20, shuffle = False):
|
| 42 |
list_label = [x+'.' if x[-1] != '.' else x for x in list_label]
|
| 43 |
list_label_new = list_label + [tokenizer.pad_token]* (label_num - len(list_label))
|
|
|
|
| 35 |
model = AutoModelForSequenceClassification.from_pretrained("DAMO-NLP-SG/zero-shot-classify-SSTuning-base")
|
| 36 |
|
| 37 |
text = "I love this place! The food is always so fresh and delicious."
|
|
|
|
| 38 |
list_label = ["negative","positve"]
|
| 39 |
|
| 40 |
+
list_ABC = [x for x in string.ascii_uppercase]
|
| 41 |
def add_prefix(text, list_label, label_num = 20, shuffle = False):
|
| 42 |
list_label = [x+'.' if x[-1] != '.' else x for x in list_label]
|
| 43 |
list_label_new = list_label + [tokenizer.pad_token]* (label_num - len(list_label))
|