Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,5 @@ tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-BigBird")
|
|
| 7 |
(2) Model type:
|
| 8 |
|
| 9 |
num_classes = 4 #T in [1,2,3,4]
|
|
|
|
| 10 |
model = BigBirdForSequenceClassification.from_pretrained(directory, num_labels=num_classes)
|
|
|
|
| 7 |
(2) Model type:
|
| 8 |
|
| 9 |
num_classes = 4 #T in [1,2,3,4]
|
| 10 |
+
|
| 11 |
model = BigBirdForSequenceClassification.from_pretrained(directory, num_labels=num_classes)
|