postbot/emailgen-pythia-410m-deduped

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"])

Open LLM Leaderboard Evaluation Results

Detailed results can be found here