Downloads · 30 days
21
3% of all-time downloads
Model-Solubility-prediction-650M
SaProtHub/Model-Solubility-prediction-650M
Model-Solubility-prediction-650M is a machine learning model from SaProtHub. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft.
This model is used to predict solubility of a amino acid sequence.
All-time downloads
642
Public
Repo size
31 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors31 MB · 100%
At a glance
- Library
- peft
- Access
- Public
- Created
- Aug 1, 2024
- Updated
- Oct 16, 2024
- SHA
- 4772680d
Base models
Model card
From the Hugging Face model README
Base model: westlake-repl/SaProt_650M_AF2
Model Card for Model ID
<!-- Provide a quick summary of what the model is/does. -->This model is used to predict solubility of a amino acid sequence.
Task type
protein level classification
Dataset description
The dataset is from DeepSol: a deep learning framework for sequence-based protein solubility prediction. Binary label, 1 means soluble, 0 means insoluble.
Model input type
Amino acid sequence
Performance
test_acc: 0.74
LoRA config
lora_dropout: 0.0
lora_alpha: 16
target_modules: ["query", "key", "value", "intermediate.dense", "output.dense"]
modules_to_save: ["classifier"]
Training config
class: AdamW
betas: (0.9, 0.98)
weight_decay: 0.01
learning rate: 1e-4
epoch: 1
batch size: 100
precision: 16-mixed
- Library
- peft
- Created
- Aug 1, 2024
- Updated
- Oct 16, 2024