Downloads · 30 days
11
2% of all-time downloads
prot_bert_bfd_rosetta204060aa
rampasek/prot_bert_bfd_rosetta204060aa
prot_bert_bfd_rosetta204060aa is a text classification model from rampasek. Use it when you need a label for a piece of text. It is set up for transformers.
This model is finetuned to predict Rosetta fold energy using a dataset of 300k protein sequences: 100k of 20AA, 100k of 40AA, and 100k of 60AA
All-time downloads
528
Public
Repo size
3.4 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin1.7 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Mar 29, 2022
- Updated
- Mar 29, 2022
- SHA
- f6db0f4a
Try a prompt
Model card
From the Hugging Face model README
ProtBert-BFD finetuned on Rosetta 20,40,60AA dataset
This model is finetuned to predict Rosetta fold energy using a dataset of 300k protein sequences: 100k of 20AA, 100k of 40AA, and 100k of 60AA
Current model in this repo: prot_bert_bfd-finetuned-032822_1323
Performance
- 20AA sequences (1k eval set):
Metrics: 'mae': 0.100418, 'r2': 0.989028, 'mse': 0.016266, 'rmse': 0.127537 - 40AA sequences (10k eval set):
Metrics: 'mae': 0.173888, 'r2': 0.963361, 'mse': 0.048218, 'rmse': 0.219587 - 60AA sequences (10k eval set):
Metrics: 'mae': 0.235238, 'r2': 0.930164, 'mse': 0.088131, 'rmse': 0.2968
prot_bert_bfd from ProtTrans
The starting pretrained model is from ProtTrans, trained on 2.1 billion proteins from BFD. It was trained on protein sequences using a masked language modeling (MLM) objective. It was introduced in this paper and first released in this repository.
Created by Ladislav Rampasek
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Mar 29, 2022
- Updated
- Mar 29, 2022