sentiment-tone-classifier / task_config.json
khulnasoft's picture
Add files using upload-large-folder tool
491e3e7 verified
{
"task_name": "sentiment_tone_classification",
"model_type": "multi_label_classification",
"sentiment_labels": ["positive", "neutral", "negative"],
"tone_labels": [
"formal",
"informal",
"poetic",
"dramatic",
"melancholic",
"joyful",
"satirical",
"romantic"
],
"num_sentiment_labels": 3,
"num_tone_labels": 8,
"max_length": 512,
"training": {
"num_epochs": 15,
"batch_size": 16,
"learning_rate": 2e-5,
"warmup_ratio": 0.1,
"weight_decay": 0.01
}
}