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