Update README.md

README.md

@@ -10,6 +10,8 @@ metrics:
 model-index:
 - name: results
   results: []
+datasets:
+- davanstrien/arxiv-new-datasets
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,7 +19,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # results
 
-This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on an unknown dataset.
+This model is a fine-tuned version of [answerdotai/ModernBERT-base](https://huggingface.co/answerdotai/ModernBERT-base) on the [davanstrien/arxiv-new-datasets](https://huggingface.co/datasets/davanstrien/arxiv-new-datasets) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.3220
 - Accuracy: 0.945
@@ -73,4 +75,4 @@ The following hyperparameters were used during training:
 - Transformers 4.48.2
 - Pytorch 2.6.0+cu124
 - Datasets 3.2.0
-- Tokenizers 0.21.0
+- Tokenizers 0.21.0