Commit
•
4f58b6d
1
Parent(s):
a17264a
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,28 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
---
|
| 4 |
+
|
| 5 |
+
# ESM-2 QLoRA for Binding Sites Prediction
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
## Testing for Overfitting
|
| 9 |
+
|
| 10 |
+
```python
|
| 11 |
+
Train metrics:
|
| 12 |
+
{'eval_loss': 0.17861589789390564,
|
| 13 |
+
'eval_accuracy': 0.9336392007583741,
|
| 14 |
+
'eval_precision': 0.24007189695313816,
|
| 15 |
+
'eval_recall': 0.9234520216135872,
|
| 16 |
+
'eval_f1': 0.38107489676203077,
|
| 17 |
+
'eval_auc': 0.9286608447868842,
|
| 18 |
+
'eval_mcc': 0.4519203165484902}
|
| 19 |
+
|
| 20 |
+
Test metrics:
|
| 21 |
+
{'eval_loss': 0.2265990674495697,
|
| 22 |
+
'eval_accuracy': 0.913988661430497,
|
| 23 |
+
'eval_precision': 0.1725452162312655,
|
| 24 |
+
'eval_recall': 0.8272126203209694,
|
| 25 |
+
'eval_f1': 0.28553230637278637,
|
| 26 |
+
'eval_auc': 0.8715212375759034,
|
| 27 |
+
'eval_mcc': 0.3539008454498742
|
| 28 |
+
```
|