Update README.md

README.md

@@ -7,34 +7,23 @@ metrics:
 accuracy: 0.68
 ---
 
-# Model Card for Model-Demo-35M
+Base model: westlake-repl/SaProt_35M_AF2
 
-## Description
-This is a protein **EC number classification model** based on SaProt_35M_AF2, fine-tuned with LoRA. The model can classify proteins into **6 major EC classes (EC1-EC6)**. Since there are only 31 samples for EC7 in the raw dataset, this class is excluded from training and prediction.
-
-Label mapping:
-- **Label 0**: Oxidoreductase (EC1)
-- **Label 1**: Transferase (EC2)
-- **Label 2**: Hydrolase (EC3)
-- **Label 3**: Lyase (EC4)
-- **Label 4**: Isomerase (EC5)
-- **Label 5**: Ligase (EC6)
+Task type: protein-level classification
 
+Dataset: This model classifies proteins into 6 major EC classes (EC1-EC6). EC7 was excluded due to only 31 samples available.
+To address class imbalance, Label 4 (EC5) was duplicated 2 times and Label 5 (EC6) was duplicated 1 time in the training set.
 Training data is obtained from: https://academic.oup.com/nar/article/54/D1/D643/8313833
 
-To address the **class imbalance problem** in the training set, we performed data augmentation:
-- Label 4 (EC5) samples were duplicated **2 times**
-- Label 5 (EC6) samples were duplicated **1 time**
-
-## Task type
-Protein-level Classification
-
-## Model input type
-Amino acid sequence (AA Sequence)
-
-## Dataset Distribution
+Label mapping:
+Label 0: Oxidoreductase (EC1)
+Label 1: Transferase (EC2)
+Label 2: Hydrolase (EC3)
+Label 3: Lyase (EC4)
+Label 4: Isomerase (EC5)
+Label 5: Ligase (EC6)
 
-### Training set
+Training set distribution:
 - Label 0: 1497 (28.5%)
 - Label 2: 1217 (23.2%)
 - Label 1: 1050 (19.9%)
@@ -43,7 +32,7 @@ Amino acid sequence (AA Sequence)
 - Label 5: 483 (9.2%)
 Total: 5255 samples
 
-### Validation set
+Validation set distribution:
 - Label 0: 187 (32.0%)
 - Label 2: 152 (26.0%)
 - Label 1: 131 (22.4%)
@@ -52,7 +41,7 @@ Total: 5255 samples
 - Label 5: 20 (3.4%)
 Total: 585 samples
 
-### Test set
+Test set distribution:
 - Label 0: 188 (31.8%)
 - Label 2: 153 (25.9%)
 - Label 1: 132 (22.3%)
@@ -61,22 +50,23 @@ Total: 585 samples
 - Label 5: 21 (3.5%)
 Total: 591 samples
 
-## Performance (on test set)
-- **Accuracy: 0.68**
+Model input type: Amino acid sequence
+
+Performance (on test set): 0.68 accuracy
 
-## LoRA config
-- **r:** 8
-- **lora_dropout:** 0.1
-- **lora_alpha:** 16
-- **target_modules:** ['key', 'value', 'output.dense', 'intermediate.dense', 'query']
-- **modules_to_save:** ['classifier']
+LoRA config:
+r: 8
+lora_dropout: 0.1
+lora_alpha: 16
+target_modules: ["key", "value", "output.dense", "intermediate.dense", "query"]
+modules_to_save: ["classifier"]
 
-## Training config
-- **optimizer:**
-  - **class:** AdamW
-  - **betas:** (0.9, 0.98)
-  - **weight_decay:** 0.01
-- **learning rate:** 0.0005
-- **epoch:** 25
-- **batch size:** 64
-- **precision:** 16-mixed
+Training config:
+optimizer:
+class: AdamW
+betas: (0.9, 0.98)
+weight_decay: 0.01
+learning rate: 0.0005
+epoch: 25
+batch size: 64
+precision: 16-mixed