Update README.md

README.md

@@ -15,7 +15,8 @@ This model was trained from scratch on an unknown dataset.
 
 ## Model description
 
-More information needed
+0.502831
+0.798512
 
 ## Intended uses & limitations
 
@@ -27,20 +28,23 @@ More information needed
 
 ## Training procedure
 
+Epoch	Training Loss	Validation Loss	Accuracy
+2	0.451100	0.502831	0.798512
+3	0.365500	0.576118	0.795456
+4	0.301900	0.625391	0.798512
+5	0.246600	0.835689	0.797963
+
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 5e-05
-- train_batch_size: 8
-- eval_batch_size: 8
-- seed: 42
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
-- num_epochs: 3.0
-
+    metric_for_best_model='accuracy',
+    learning_rate=2e-5,
+    num_train_epochs=20,
+    weight_decay=0.01,
+    per_device_train_batch_size=16,   # batch size per device during training
+    per_device_eval_batch_size=16,    # batch size for evaluation
+    evaluation_strategy="epoch",
+    save_strategy="epoch",
+    save_total_limit = 2,
+    load_best_model_at_end=True)
 ### Framework versions
-
-- Transformers 4.36.0
-- Pytorch 2.1.0+cu118
-- Datasets 2.11.0
-- Tokenizers 0.15.0