CancerStage_Classifier_M

作者 jkefeli · 下载 2 · 点赞 0 · 更新 2023-11-02

To use the model, add the following from the transformers package:

(1) Tokenizer:

tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-BigBird")

(2) Model type:

#M in [0,1]

num_classes = 2

model = BigBirdForSequenceClassification.from_pretrained(directory, num_labels=num_classes)