Update README.md

README.md

@@ -1,19 +1,44 @@
 ---
+license: apache-2.0
 tags:
-- generated_from_trainer
+- CodeGPT-small-py
+- hearthstone
 metrics:
 - bleu
+- dvitel/codebleu
+- exact_match
+- chrf
+datasets:
+- dvitel/hearthstone
 model-index:
 - name: h0-1
-  results: []
----
+  results:
+  - task:
+      type: text-generation
+      name: Python Code Synthesis
+    dataset:
+      type: dvitel/hearthstone
+      name: HearthStone
+      split: test
+    metrics:
+      - type: exact_match
+        value: 0.21212121212121213
+        name: Exact Match
+      - type: bleu
+        value: 0.8954467480979604
+        name: BLEU        
+      - type: dvitel/codebleu
+        value: 0.6976253554171774
+        name: CodeBLEU                
+      - type: dvitel/codebleu
+        value: 91.42413429212283
+        name: chrF                        
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
+---
 
 # h0-1
 
-This model is a fine-tuned version of [microsoft/CodeGPT-small-py](https://huggingface.co/microsoft/CodeGPT-small-py) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/CodeGPT-small-py](https://huggingface.co/microsoft/CodeGPT-small-py) on [hearthstone](https://huggingface.co/datasets/dvitel/hearthstone) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.3622
 - Exact Match: 0.1970
@@ -23,15 +48,15 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+CodeGPT-small-py fine-tuned on HearthStone dataset for 200 epochs
 
 ## Intended uses & limitations
 
-More information needed
+HearthStone card code synthesis. 
 
 ## Training and evaluation data
 
-More information needed
+See split of [hearthstone](https://huggingface.co/datasets/dvitel/hearthstone) dataset
 
 ## Training procedure