Update README.md

README.md

@@ -5,7 +5,7 @@ base_model: xlm-roberta-large
 tags:
 - generated_from_trainer
 datasets:
-- generator
+- stulcrad/CERED-2
 metrics:
 - accuracy
 model-index:
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 # XLM-RoBERTa-2
 
-This model is a fine-tuned version of [xlm-roberta-large](https://huggingface.co/xlm-roberta-large) on the generator dataset.
-It achieves the following results on the evaluation set:
+This model is a fine-tuned version of [xlm-roberta-large](https://huggingface.co/xlm-roberta-large) on the Czech Relationship Extraction Dataset (http://hdl.handle.net/11234/1-3265).
+It achieves the following results on the test set:
 - Loss: 0.6614
 - Accuracy: 0.9100
 - Micro Precision: 0.9100
@@ -76,4 +76,4 @@ The following hyperparameters were used during training:
 - Transformers 4.46.2
 - Pytorch 2.5.1+cu124
 - Datasets 3.1.0
-- Tokenizers 0.20.3
+- Tokenizers 0.20.3