ymoslem commited on
Commit
5ce0445
·
verified ·
1 Parent(s): 240650e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -4
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
+ ```