Update README.md
Browse files
README.md
CHANGED
@@ -36,7 +36,7 @@ All training data and scripts are available at the [GitHub repository](https://g
|
|
36 |
|
37 |
### Training Procedure
|
38 |
|
39 |
-
The model was fine-tuned for a regression task using a Mean Squared Error (MSE) loss function. Key hyperparameters include a learning rate of 5e-5, a batch size of
|
40 |
|
41 |
### Evaluation Results
|
42 |
|
@@ -44,8 +44,8 @@ The model's performance was evaluated on two separate, unseen test sets. The res
|
|
44 |
|
45 |
| Test Set | Pearson Correlation | Spearman Correlation |
|
46 |
|-----------------|---------------------|----------------------|
|
47 |
-
| New Testament | 0.
|
48 |
-
| Homeric | 0.
|
49 |
|
50 |
|
51 |
## How to Use
|
|
|
36 |
|
37 |
### Training Procedure
|
38 |
|
39 |
+
The model was fine-tuned for a regression task using a Mean Squared Error (MSE) loss function. Key hyperparameters include a learning rate of 5e-5, a batch size of 64, and an AdamW optimizer. Training was performed with early stopping based on the Spearman correlation on a validation set.
|
40 |
|
41 |
### Evaluation Results
|
42 |
|
|
|
44 |
|
45 |
| Test Set | Pearson Correlation | Spearman Correlation |
|
46 |
|-----------------|---------------------|----------------------|
|
47 |
+
| New Testament | 0.661 | 0.648 |
|
48 |
+
| Homeric | 0.660 | 0.649 |
|
49 |
|
50 |
|
51 |
## How to Use
|