Update README.md

README.md

@@ -5,18 +5,40 @@ library_name: peft
 ## Training procedure
 
 Generalization capabilities based on [this notebook](https://huggingface.co/AmelieSchreiber/esm2_t12_35M_lora_binding_sites_770K_v1/blob/main/esmbind-validation-struct.ipynb)
-are still lacking. The average metrics on the train datasets from the paper linked to in the notebook are:
+are still lacking. 
 
+Test Metrics:
+
+```python
+'eval_loss': 0.4365999102592468,
+'eval_accuracy': 0.9420512498988455,
+'eval_precision': 0.2294216317399737,
+'eval_recall': 0.7584797572261781,
+'eval_f1': 0.35228549223252686,
+'eval_auc': 0.8522129916967822,
+'eval_mcc': 0.39725936154292724,
 ```
+
+The metrics on the train and test datasets from the paper mentioned in the notebook are:
+
+```python
+Train metrics: 
 Average Accuracy: 0.9169
 Average Precision: 0.1416
 Average Recall: 0.2769
 Average F1 Score: 0.1743
 Average AUC: 0.6088
 Average MCC: 0.1522
+
+Test metrics: 
+Average Accuracy: 0.9199
+Average Precision: 0.1307
+Average Recall: 0.2750
+Average F1 Score: 0.1677
+Average AUC: 0.6081
+Average MCC: 0.1474
 ```
 
-The average metrics on the test datasets provided by the paper are very similar. 
 
 ### Framework versions