Update README.md

README.md

@@ -6,7 +6,32 @@ datasets:
 - arxiv-summarization
 model-index:
 - name: Paper-Summarization-ArXiv
-  results: []
+  results:
+  - task:
+      name: Summarization
+      type: summarization
+    dataset:
+      name: ccdv/arxiv-summarization
+      type: ccdv/arxiv-summarization
+      config: section
+      split: test
+      args: section
+    metrics:
+    - name: ROUGE-1
+      type: rouge
+      value: 43.2305
+    - name: ROUGE-2
+      type: rouge
+      value: 16.6571
+    - name: ROUGE-L
+      type: rouge
+      value: 24.4315      
+    - name: ROUGE-LSum
+      type: rouge
+      value: 33.9399
+    
+
+
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,9 +40,22 @@ should probably proofread and complete it, then remove this comment. -->
 # Paper-Summarization-ArXiv
 
 This model is a fine-tuned version of [google/pegasus-x-base](https://huggingface.co/google/pegasus-x-base) on the arxiv-summarization dataset.
+
 It achieves the following results on the evaluation set:
 - Loss: 2.0127
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss |
+|:-------------:|:-----:|:-----:|:---------------:|
+| 2.6153        | 1.0   | 3172  | 2.1045          |
+| 2.202         | 2.0   | 6344  | 2.0511          |
+| 2.1547        | 3.0   | 9516  | 2.0282          |
+| 2.132         | 4.0   | 12688 | 2.0164          |
+| 2.1222        | 5.0   | 15860 | 2.0127          |
+
+
+
 ## Model description
 
 More information needed
@@ -28,13 +66,14 @@ Paper Summarization
 
 ## Compare to Baseline
 - **Pegasus-X-base Zero-shot Performance:**
-  - 'ROUGE-1': 0.062269
-  - 'ROUGE-2': 0.007894
-  - 'ROUGE-L': 0.046905
-  - 'ROUGE-Lsum': 0.054591
+  - ROUGE-1 | ROUGE-2 | ROUGE-L | ROUGE-LSUM : 6.2269 | 0.7894 | 4.6905 | 5.4591
 - **Our Model (Generated with length_penalty=1, num_beams=2, max_length=128*4,min_length=150, no_repeat_ngram_size= 3, top_k=25,top_p=0.95**)
-  - {'rouge1': 0.43230484397753544, 'rouge2': 0.16657053834996538, 'rougeL': 0.2443145883290619, 'rougeLsum': 0.3393990490342812}
-- **Our Model**
+  - ROUGE-1 | ROUGE-2 | ROUGE-L | ROUGE-LSUM : 43.2305 | 16.6571 | 24.4315 | 33.9399
+    
+
+
+
+
 ## Training and evaluation data
 
 More information needed
@@ -55,16 +94,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_steps: 1586
 - num_epochs: 5
 
-### Training results
-
-| Training Loss | Epoch | Step  | Validation Loss |
-|:-------------:|:-----:|:-----:|:---------------:|
-| 2.6153        | 1.0   | 3172  | 2.1045          |
-| 2.202         | 2.0   | 6344  | 2.0511          |
-| 2.1547        | 3.0   | 9516  | 2.0282          |
-| 2.132         | 4.0   | 12688 | 2.0164          |
-| 2.1222        | 5.0   | 15860 | 2.0127          |
-
 
 ### Framework versions