AmelieSchreiber
commited on
Commit
•
3ae8a1d
1
Parent(s):
fc8b744
Update README.md
Browse files
README.md
CHANGED
@@ -17,5 +17,21 @@ tags:
|
|
17 |
## Testing for Overfitting
|
18 |
|
19 |
```python
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
```
|
|
|
17 |
## Testing for Overfitting
|
18 |
|
19 |
```python
|
20 |
+
Train metrics:
|
21 |
+
{'eval_loss': 0.09572703391313553,
|
22 |
+
'eval_accuracy': 0.9670769479865963,
|
23 |
+
'eval_precision': 0.3970221190232079,
|
24 |
+
'eval_recall': 0.9411011487595375,
|
25 |
+
'eval_f1': 0.5584507515735834,
|
26 |
+
'eval_auc': 0.9543828770020467,
|
27 |
+
'eval_mcc': 0.5996252550053665}
|
28 |
|
29 |
+
Test metrics:
|
30 |
+
{'eval_loss': 0.1680256575345993,
|
31 |
+
'eval_accuracy': 0.943313091525589,
|
32 |
+
'eval_precision': 0.2342637814982173,
|
33 |
+
'eval_recall': 0.7618306193745306,
|
34 |
+
'eval_f1': 0.35833816875074714,
|
35 |
+
'eval_auc': 0.8544971814140561,
|
36 |
+
'eval_mcc': 0.40290081143832884}
|
37 |
```
|