Update README.md
Browse files
README.md
CHANGED
@@ -14,18 +14,18 @@ Used half of the train dataset
|
|
14 |
|
15 |
# Training parameters
|
16 |
|
17 |
-
2 epochs
|
18 |
-
train batch size = 8
|
19 |
-
eval batch size = 8
|
20 |
-
gradient accumulation steps = 4
|
21 |
-
learning rate = 5e-5 without warm up and decay
|
22 |
|
23 |
# Metrics
|
24 |
|
25 |
-
Achieved
|
26 |
-
accuracy = 0.86
|
27 |
-
balanced = 0.76
|
28 |
-
macro f1 score = 0.81
|
29 |
on test set, improving accucary and f1 score compared to dataset baseline
|
30 |
|
31 |
# Usage
|
|
|
14 |
|
15 |
# Training parameters
|
16 |
|
17 |
+
- 2 epochs
|
18 |
+
- train batch size = 8
|
19 |
+
- eval batch size = 8
|
20 |
+
- gradient accumulation steps = 4
|
21 |
+
- learning rate = 5e-5 without warm up and decay
|
22 |
|
23 |
# Metrics
|
24 |
|
25 |
+
Achieved
|
26 |
+
- accuracy = 0.86
|
27 |
+
- balanced = 0.76
|
28 |
+
- macro f1 score = 0.81
|
29 |
on test set, improving accucary and f1 score compared to dataset baseline
|
30 |
|
31 |
# Usage
|