emailgen-pythia-410m-deduped

This model is a fine-tuned version of EleutherAI/pythia-410m-deduped on email data. It achieves the following results on the evaluation set:

• Loss: 2.1018
• Accuracy: 0.6157
• perplexity: 8.181

Model description

• fine-tuned on dataset of emails for 4 epochs
• intended use: "text completion" of partially written emails

Usage example

from transformers import pipeline

model_tag = "postbot/emailgen-pythia-410m-deduped"
generator = pipeline(
    "text-generation",
    model=model_tag,
)

prompt = """
Hello, 

Following up on the bubblegum shipment."""

result = generator(
    prompt,
)  # generate
print(result[0]["generated_text"])