jkefeli commited on
Commit
45602a9
·
1 Parent(s): 77c86f8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -6,6 +6,9 @@ tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-BigBird")
6
 
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)
 
6
 
7
  (2) Model type:
8
 
9
+ #T in [1,2,3,4]
10
+
11
+
12
+ num_classes = 4
13
 
14
  model = BigBirdForSequenceClassification.from_pretrained(directory, num_labels=num_classes)