Commit
·
5d1e746
1
Parent(s):
95a2713
Update README.md
Browse files
README.md
CHANGED
|
@@ -37,4 +37,26 @@ Test (epoch 1):
|
|
| 37 |
```
|
| 38 |
|
| 39 |
The metrics on the datasets [mentioned here](https://github.com/hamzagamouh/pt-lm-gnn)
|
| 40 |
-
can be [found here](https://huggingface.co/AmelieSchreiber/esm2_t6_8m_qlora_binding_sites_v1/blob/main/pdb_structure_metrics.txt).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
```
|
| 38 |
|
| 39 |
The metrics on the datasets [mentioned here](https://github.com/hamzagamouh/pt-lm-gnn)
|
| 40 |
+
can be [found here](https://huggingface.co/AmelieSchreiber/esm2_t6_8m_qlora_binding_sites_v1/blob/main/pdb_structure_metrics.txt).
|
| 41 |
+
|
| 42 |
+
## Testing for Overfitting
|
| 43 |
+
|
| 44 |
+
```python
|
| 45 |
+
Train metrics:
|
| 46 |
+
{'eval_loss': 0.35603779554367065,
|
| 47 |
+
'eval_accuracy': 0.8439650327744697,
|
| 48 |
+
'eval_precision': 0.11529132737114746,
|
| 49 |
+
'eval_recall': 0.9162279099673907,
|
| 50 |
+
'eval_f1': 0.20481078411524478,
|
| 51 |
+
'eval_auc': 0.8792862815250805,
|
| 52 |
+
'eval_mcc': 0.29286338236467047}
|
| 53 |
+
|
| 54 |
+
Test metrics:
|
| 55 |
+
{'eval_loss': 0.3942357003688812,
|
| 56 |
+
'eval_accuracy': 0.8246741787222583,
|
| 57 |
+
'eval_precision': 0.0942294455869611,
|
| 58 |
+
'eval_recall': 0.8169195154212542,
|
| 59 |
+
'eval_f1': 0.16896879944226734,
|
| 60 |
+
'eval_auc': 0.8208833317810486,
|
| 61 |
+
'eval_mcc': 0.23939865094539936}
|
| 62 |
+
```
|