YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
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 |
- Downloads last month
- 3