Update README.md

README.md

@@ -13,12 +13,27 @@ should probably proofread and complete it, then remove this comment. -->
 
 # DSPFirst-Finetuning-3
 
-This model is a fine-tuned version of [ahotrod/electra_large_discriminator_squad2_512](https://huggingface.co/ahotrod/electra_large_discriminator_squad2_512) on an unknown dataset.
+This model is a fine-tuned version of [ahotrod/electra_large_discriminator_squad2_512](https://huggingface.co/ahotrod/electra_large_discriminator_squad2_512) on a generated Questions and Answers dataset from the DSPFirst textbook based on the SQuAD 2.0 format.
 It achieves the following results on the evaluation set:
 - Loss: 0.9996
 - Exact: 63.9193
 - F1: 72.1090
 
+# Dataset
+A visualization of the dataset can be found [here](https://github.gatech.edu/pages/VIP-ITS/textbook_SQuAD_explore/explore/textbookv1.0/textbook/). The split between train and test is 80% and 20% respectively.
+```
+DatasetDict({
+    train: Dataset({
+        features: ['id', 'title', 'context', 'question', 'answers'],
+        num_rows: 4755
+    })
+    test: Dataset({
+        features: ['id', 'title', 'context', 'question', 'answers'],
+        num_rows: 1189
+    })
+})
+```
+
 ## Model description
 
 More information needed
@@ -46,6 +61,10 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 3
 
+### Model hyperparameters
+- hidden_dropout_prob: 0.35
+- attention_probs_dropout_prob = 0.35
+
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Exact   | F1      |