luvnpce83 commited on
Commit
ca58545
·
verified ·
1 Parent(s): 07522d3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 32, 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,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.643 | 0.629 |
48
- | Homeric | 0.639 | 0.628 |
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