Update README.md
Browse files
README.md
CHANGED
@@ -53,7 +53,10 @@ model-index:
|
|
53 |
- name: Root Mean Squared Error
|
54 |
type: RMSE
|
55 |
value: 0.2375
|
56 |
-
|
|
|
|
|
|
|
57 |
---
|
58 |
|
59 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -67,10 +70,16 @@ on the [ymoslem/wmt-da-human-evaluation](https://huggingface.co/ymoslem/wmt-da-h
|
|
67 |
It achieves the following results on the evaluation set:
|
68 |
- Loss: 0.0564
|
69 |
|
|
|
|
|
|
|
|
|
70 |
## Training procedure
|
71 |
|
72 |
### Training hyperparameters
|
73 |
|
|
|
|
|
74 |
The following hyperparameters were used during training:
|
75 |
- learning_rate: 8e-05
|
76 |
- train_batch_size: 128
|
@@ -223,6 +232,4 @@ with torch.no_grad():
|
|
223 |
|
224 |
# Extend the list with the predictions
|
225 |
predictions.extend(batch_predictions.tolist())
|
226 |
-
```
|
227 |
-
|
228 |
-
|
|
|
53 |
- name: Root Mean Squared Error
|
54 |
type: RMSE
|
55 |
value: 0.2375
|
56 |
+
metrics:
|
57 |
+
- pearsonr
|
58 |
+
- mae
|
59 |
+
- r_squared
|
60 |
---
|
61 |
|
62 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
70 |
It achieves the following results on the evaluation set:
|
71 |
- Loss: 0.0564
|
72 |
|
73 |
+
## Model description
|
74 |
+
|
75 |
+
This model is for reference-free quality estimation (QE) of machine translation (MT) systems.
|
76 |
+
|
77 |
## Training procedure
|
78 |
|
79 |
### Training hyperparameters
|
80 |
|
81 |
+
This version of the model uses `tokenizer.model_max_length` 512
|
82 |
+
|
83 |
The following hyperparameters were used during training:
|
84 |
- learning_rate: 8e-05
|
85 |
- train_batch_size: 128
|
|
|
232 |
|
233 |
# Extend the list with the predictions
|
234 |
predictions.extend(batch_predictions.tolist())
|
235 |
+
```
|
|
|
|