Downloads · 30 days
14
3% of all-time downloads
biobert-finetuned-genetic-mutation
wahdan99/biobert-finetuned-genetic-mutation
biobert-finetuned-genetic-mutation is a text classification model from wahdan99. Use it when you need a label for a piece of text. It is set up for transformers.
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 b…
All-time downloads
471
Public
Repo size
867 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin433 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Feb 28, 2023
- Updated
- Feb 28, 2023
- SHA
- 1d06a641
Try a prompt
Model card
From the Hugging Face model README
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
<table style="width:100%" border='1 solid black'> <tr> <th>evaluation metric</th> <th>evaluation value</th> </tr> <tr> <td>train loss</td> <td>0.639870</td> </tr> <tr> <td>validation loss</td> <td>0.824237</td> </tr> <tr> <td>f1 score</td> <td>0.733120</td> </tr> <tr> <td>accuracy</td> <td>0.728585</td> </tr> </table>- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Feb 28, 2023
- Updated
- Feb 28, 2023