lucasrct's picture
Upload folder using huggingface_hub
32e821b verified

sentiment_prompt_2_model

Model Description

This model is a fine-tuned version of EleutherAI/pythia-410m trained on dair-ai/emotion data.

Dataset Details

  • Dataset Configuration: unsplit
  • Dataset Name: dair-ai/emotion
  • Prompt: text: {text} sentiment: {label}

Training Details

  • Base Model: EleutherAI/pythia-410m
  • Training Parameters:
    • Learning Rate: 2e-05
    • Batch Size: 1
    • Epochs: 1