| pipeline_tag: text-classification | |
| library_name: transformers | |
| tags: | |
| - emotion-classification | |
| - tone-mapping | |
| - tonepilot | |
| - bert | |
| language: | |
| - en | |
| # TonePilot BERT Classifier | |
| This model maps input emotional tones to appropriate response personalities for the TonePilot system. | |
| ## Model Details | |
| - **Base Model**: roberta-base | |
| - **Task**: Multi-label emotion/tone classification | |
| - **Labels**: 73 response personality types | |
| - **Training**: Custom dataset for emotional tone mapping | |
| ## Usage | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline( | |
| "text-classification", | |
| model="sdurgi/bert_emotion_response_classifier", | |
| return_all_scores=True | |
| ) | |
| # Input: detected emotions from text | |
| result = classifier("curious, confused") | |
| print(result) | |
| ``` | |
| ## Labels | |
| analytical, angry, anxious, apologetic, appreciative, calm_coach, calming, casual, cautious, celebratory, cheeky, clear, compassionate, compassionate_friend, complimentary, confident, confident_flirt, confused, congratulatory, curious, direct, direct_ally, directive, empathetic, empathetic_listener, encouraging, engaging, enthusiastic, excited, flirty, friendly, gentle, gentle_mentor, goal_focused, helpful, hopeful, humorous, humorous (lightly), informative, inquisitive, insecure, intellectual, joyful, light-hearted, light-humored, lonely, motivational_coach, mysterious, nurturing_teacher, overwhelmed, patient, personable, playful, playful_partner, practical_dreamer, problem-solving, realistic, reassuring, resourceful, sad, sarcastic, sarcastic_friend, speculative, strategic, suggestive, supportive, thoughtful, tired, upbeat, validating, warm, witty, zen_mirror | |
| ## Integration | |
| This model is designed to work with the TonePilot system: | |
| 1. Input text → HF emotion tagger detects emotions | |
| 2. Detected emotions → This model maps to response personalities | |
| 3. Response personalities → Prompt builder creates contextual prompts | |