Update README.md

README.md

@@ -6,7 +6,7 @@ datasets:
 metrics:
 - accuracy
 model-index:
-- name: estbert128_lr5e-5_b16_s3
+- name: EstBERT128_Rubric
   results:
   - task:
       name: Text Classification
@@ -24,9 +24,9 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# estbert128_lr5e-5_b16_s3
+# EstBERT128_Rubric
 
-This model is a fine-tuned version of [tartuNLP/EstBERT](https://huggingface.co/tartuNLP/EstBERT) on the rubric dataset.
+This model is a fine-tuned version of [tartuNLP/EstBERT](https://huggingface.co/tartuNLP/EstBERT) on the rubric categories of the [Estonian Valence dataset](http://peeter.eki.ee:5000/valence/paragraphsquery).
 It achieves the following results on the evaluation set:
 - Loss: 2.0552
 - Accuracy: 0.8329