Hugging Face's logo

jkefeli
/
CancerStage_Classifier_T

Text Classification
Transformers
PyTorch
big_bird
Model card Files
xet
 Community
1
jkefeli commited on
Commit
2b622e6
·
1 Parent(s): 6cf7502

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -7,4 +7,4 @@ 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("yikuan8/Clinical-BigBird", num_labels=num_classes)
 
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)