Update README.md

README.md

@@ -7,60 +7,93 @@ metrics:
 model-index:
 - name: multi-emails-hq-pythia-410m-deduped-r1
   results: []
+widget:
+  - text: >-
+      Good Morning Professor Beans,
+
+      Hope you are doing well. I just wanted to reach out and ask if
+      differential calculus will be on the exam
+    example_title: email to prof
+  - text: >-
+      Hey <NAME>,
+
+
+      Thank you for signing up for my weekly newsletter. Before we get started,
+      you'll have to confirm your email address.
+    example_title: newsletter
+  - text: >-
+      Hi <NAME>,
+
+
+      I hope this email finds you well. I wanted to reach out and ask about
+      office hours
+    example_title: office hours
+  - text: >-
+      Greetings <NAME>,
+
+
+      I hope you had a splendid evening at the Company sausage eating festival.
+      I am reaching out because
+    example_title: festival
+  - text: |-
+      Good Morning Harold,
+
+      I was wondering when the next
+    example_title: event
+  - text: URGENT - I need the TPS reports
+    example_title: URGENT
+  - text: |-
+      Hi Archibald,
+
+      I hope this email finds you extremely well.
+    example_title: emails that find you
+  - text: |-
+      Hello there.
+
+      I just wanted to reach out and check in to
+    example_title: checking in
+  - text: >-
+      Hello <NAME>,
+
+
+      I hope this email finds you well. I wanted to reach out and see if you've
+      enjoyed your time with us
+    example_title: work well
+  - text: >-
+      Hi <NAME>,
+
+
+      I hope this email finds you well. I wanted to reach out and see if we
+      could catch up
+    example_title: catch up
+  - text: >-
+      I'm <NAME> and I just moved into the area and wanted to reach out and get
+      some details on where I could get groceries and
+    example_title: grocery
+datasets:
+  - postbot/multi-emails-hq
+language:
+  - en
+pipeline_tag: text-generation
 ---
 
-<!-- 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. -->
 
-# multi-emails-hq-pythia-410m-deduped-r1
+# emailgen-pythia-410m-deduped
 
 This model is a fine-tuned version of [EleutherAI/pythia-410m-deduped](https://huggingface.co/EleutherAI/pythia-410m-deduped) on the None dataset.
 It achieves the following results on the evaluation set:
 - Loss: 2.1018
 - Accuracy: 0.6157
+- perplexity: 8.181
 
+- 
 ## Model description
 
-More information needed
+- fine-tuned on dataset of emails for 4 epochs
+- intended use: "text completion" of partially written emails
 
-## Intended uses & limitations
 
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
-
-### Training hyperparameters
-
-The following hyperparameters were used during training:
-- learning_rate: 0.0002
-- train_batch_size: 8
-- eval_batch_size: 2
-- seed: 69
-- distributed_type: multi-GPU
-- gradient_accumulation_steps: 16
-- total_train_batch_size: 128
-- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: cosine
-- lr_scheduler_warmup_ratio: 0.05
-- num_epochs: 4.0
-
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.3067        | 1.0   | 38   | 2.3559          | 0.5594   |
-| 1.9243        | 2.0   | 76   | 2.1283          | 0.5975   |
-| 1.5971        | 3.0   | 114  | 2.0759          | 0.6140   |
-| 1.4139        | 4.0   | 152  | 2.1018          | 0.6157   |
 
+---
 
-### Framework versions
 
-- Transformers 4.27.0.dev0
-- Pytorch 2.0.0.dev20230130+cu118
-- Datasets 2.9.0
-- Tokenizers 0.13.2