Update README.md

README.md

@@ -1,39 +1,49 @@
+base_model: westlake-repl/SaProt_35M_AF2
 
----
+Model Description
 
-base_model: westlake-repl/SaProt_35M_AF2 
+This model is fine-tuned to predict the mutation effects of *Bacillus subtilis* α-amylase. It takes amino acid sequences as input and performs protein-level regression to predict quantitative enzyme activity values, enabling accurate assessment of how mutations alter enzyme function.
 
-library_name: peft
+Task type: protein-level regression
 
----
+Model input type: Amino acid sequence
 
+Dataset
 
+The dataset is sourced from van der Flier et al. (2024), available at https://www.sciencedirect.com/science/article/pii/S2001037024002940. It contains a total of 3706 rows, with each sample carrying 1 to 8 mutation sites. The full dataset is randomly split into training, validation, and test sets following an 8:1:1 ratio. The target label is absorbance. A higher absorbance value indicates greater starch degradation range from(), corresponding to stronger detergent activity of the amylase enzyme.
 
-# Model Card for Model-Demo-35M
-<slot name='description'>
+Performance (on test set)
 
-## Task type
-Protein-level Regression
+Spearman correlation: 0.76
 
-## Model input type
-AA Sequence
+Coefficient of determination (R²): 0.69
 
-## LoRA config
+LoRA config
 
-- **r:** 8
-- **lora_dropout:** 0.1
-- **lora_alpha:** 16
-- **target_modules:** ['query', 'intermediate.dense', 'value', 'output.dense', 'key']
-- **modules_to_save:** ['classifier']
+r: 8
 
-## Training config
+lora_dropout: 0.1
 
-- **optimizer:**
-  - **class:** AdamW
-  - **betas:** (0.9, 0.98)
-  - **weight_decay:** 0.01
-- **learning rate:** 0.0005
-- **epoch:** 30
-- **batch size:** 64
-- **precision:** 16-mixed 
+lora_alpha: 16
 
+target_modules: ["query", "intermediate.dense", "value", "output.dense", "key"]
+
+modules_to_save: ["classifier"]
+
+Training config
+
+optimizer:
+
+class: AdamW
+
+betas: (0.9, 0.98)
+
+weight_decay: 0.01
+
+learning rate: 0.0005
+
+epoch: 30
+
+batch size: 64
+
+precision: 16-mixed