Update README.md

README.md

@@ -5,6 +5,9 @@ tags:
 model-index:
 - name: xtremedistil-l6-h256-uncased-question-vs-statement-classifier
   results: []
+datasets:
+- jonaskoenig/Questions-vs-Statements-Classification
+
 ---
 
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
@@ -12,7 +15,8 @@ probably proofread and complete it, then remove this comment. -->
 
 # xtremedistil-l6-h256-uncased-question-vs-statement-classifier
 
-This model is a fine-tuned version of [microsoft/xtremedistil-l6-h256-uncased](https://huggingface.co/microsoft/xtremedistil-l6-h256-uncased) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/xtremedistil-l6-h256-uncased](https://huggingface.co/microsoft/xtremedistil-l6-h256-uncased) on [question-vs-statement-classifier](https://huggingface.co/datasets/jonaskoenig/Questions-vs-Statements-Classification) dataset, which is a clone of the keggle [Questions vs Statements Classification](https://www.kaggle.com/datasets/shahrukhkhan/questions-vs-statementsclassificationdataset) dataset.
+
 It achieves the following results on the evaluation set:
 - Train Loss: 0.0227
 - Train Sparse Categorical Accuracy: 0.9894