Update README.md
Browse files
README.md
CHANGED
@@ -23,15 +23,17 @@ It achieves the following results on the evaluation set:
|
|
23 |
|
24 |
## Model description
|
25 |
|
26 |
-
|
27 |
|
28 |
-
|
|
|
29 |
|
30 |
-
|
|
|
31 |
|
32 |
## Training and evaluation data
|
33 |
|
34 |
-
|
35 |
|
36 |
## Training procedure
|
37 |
|
|
|
23 |
|
24 |
## Model description
|
25 |
|
26 |
+
“The parameters are as follows:
|
27 |
|
28 |
+
```python
|
29 |
+
{'num_train_epoch': 9, 'alpha': 0.16484386886358915, 'temperature': 2}
|
30 |
|
31 |
+
These were obtained using grid search with Optuna.
|
32 |
+
```
|
33 |
|
34 |
## Training and evaluation data
|
35 |
|
36 |
+
Clinic dataset from 'datasets'
|
37 |
|
38 |
## Training procedure
|
39 |
|