Update README.md

README.md

@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # sentiment-polish-gpt2-large
 
-This model is a fine-tuned version of [sdadas/polish-gpt2-large](https://huggingface.co/sdadas/polish-gpt2-large) on the None dataset.
+This model is a fine-tuned version of [sdadas/polish-gpt2-large](https://huggingface.co/sdadas/polish-gpt2-large) on the [polemo2-official](https://huggingface.co/datasets/clarin-pl/polemo2-official) dataset.
 It achieves the following results on the evaluation set:
 - epoch: 10.0
 - eval_accuracy: 0.9634
@@ -30,24 +30,42 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+Trained from [polish-gpt2-large](https://huggingface.co/sdadas/polish-gpt2-large)
 
 ## Intended uses & limitations
 
-More information needed
+Sentiment analysis - neutral/negative/positive/ambiguous
 
 ## Training and evaluation data
 
-More information needed
+Merged all rows from [polemo2-official](https://huggingface.co/datasets/clarin-pl/polemo2-official) dataset.
+
+Discarded rows with length > 512.
+
+Train/test split: 80%/20%
+
+Datacollator:
+```py
+data_collator = DataCollatorWithPadding(
+  tokenizer=tokenizer,
+  padding="longest",
+  max_length=MAX_INPUT_LENGTH,
+  pad_to_multiple_of=8
+)
+```
 
 ## Training procedure
 
+GPU: 2x RTX 4060Ti 16GB
+
+Training time: 29:16:50
+
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 2
-- eval_batch_size: 2
+- train_batch_size: 1
+- eval_batch_size: 1
 - seed: 42
 - gradient_accumulation_steps: 8
 - total_train_batch_size: 16