Skip to content

Commit 4077d61

Browse files
authored
Merge pull request #130 from SFI-Visual-Intelligence/solveig_results
added results for MagnusModel and SVHN
2 parents a03a6c3 + c0774bf commit 4077d61

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

README.md

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,25 @@ This section reports the results from using the model "JanModel" and the dataset
5454
For this experiment we use all five available metrics, and train for a total of 20 epochs.
5555

5656
We achieve a great fit on the data. Below are the results for the described run:
57+
5758
| Dataset Split | Loss | Entropy | Accuracy | Precision | Recall | F1 |
5859
|---------------|-------|---------|----------|-----------|--------|-------|
5960
| Train | 0.000 | 0.000 | 1.000 | 1.000 | 1.000 | 1.000 |
6061
| Validation | 0.035 | 0.006 | 0.991 | 0.991 | 0.991 | 0.991 |
61-
| Test | 0.024 | 0.004 | 0.994 | 0.994 | 0.994 | 0.994 |
62+
| Test | 0.024 | 0.004 | 0.994 | 0.994 | 0.994 | 0.994 |
63+
64+
65+
## MagnusModel & SVHN
66+
The MagnusModel was trained on the SVHN dataset, utilizing all five metrics.
67+
Employing micro-averaging for the calculation of F1 score, accuracy, recall, and precision, the model was fine-tuned over 20 epochs.
68+
A learning rate of 0.001 and a batch size of 64 were selected to optimize the training process.
69+
70+
The table below presents the detailed results, showcasing the model's performance across these metrics.
71+
72+
73+
| Dataset Split | Loss | Entropy | Accuracy | Precision | Recall | F1 |
74+
|---------------|-------|---------|----------|-----------|--------|-------|
75+
| Train | 1.007 | 0.998 | 0.686 | 0.686 | 0.686 | 0.686 |
76+
| Validation | 1.019 | 0.995 | 0.680 | 0.680 | 0.680 | 0.680 |
77+
| Test | 1.196 | 0.985 | 0.634 | 0.634 | 0.634 | 0.634 |
78+

0 commit comments

Comments
 (0)