wahdan99/biobert-finetuned-genetic-mutation

biobert finetuned for genetic mutation

In this model I have finetuned the biobert model on the Personalized Medicine: Redefining Cancer Treatment kaggle dataset which is a clinical text dataset, and There are nine different classes a genetic mutation can be classified into.

How to use this model

from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("wahdan99/biobert-finetuned-genetic-mutation")

model = AutoModelForSequenceClassification.from_pretrained("wahdan99/biobert-finetuned-genetic-mutation")

Evaluation

evaluation metric	evaluation value
train loss	0.639870
validation loss	0.824237
f1 score	0.733120
accuracy	0.728585