Classificação da emoção

O modelo de emoção no workflow de classificação do Watson Natural Language Processing classifica a emoção no texto de entrada.

Nome do fluxo de trabalho

• emotion_aggregated-workflow_en_stock
• emotion_aggregated-workflow_fr_stock

Idiomas suportados

• Inglês e francês

Recursos

O modelo de classificação de emoção é um modelo de classificação de documentos pré-treinado para a tarefa de classificação de emoção no documento de entrada. O modelo identifica a emoção de um documento e classifica-a como:

• Raiva
• Repugnância
• Medo
• Alegria
• Tristeza

Ao contrário do modelo de sentimento, que classifica cada sentença individual, o modelo de emoção classifica o documento de entrada inteiro. Como tal, o modelo de emoção funciona de forma otimizada quando o texto de entrada a ser classificado não tem mais que 1.000 caracteres. Se você gostaria de classificar textos com mais de 1.000 caracteres, divida o texto em sentenças ou parágrafos, por exemplo, e aplique o modelo de emoção a cada sentença ou parágrafo.

Um documento pode ser classificado em várias categorias ou em nenhuma categoria.

Dependências de outros blocos

Nenhuma

Código de amostra

import watson_nlp

# Load the Emotion workflow model for English
emotion_model = watson_nlp.load('emotion_aggregated-workflow_en_stock')

# Run the Tone model
emotion_result = emotion_model.run("I'm so annoyed that this code won't run")
print(emotion_result)

Saída da amostra de código:

{
  "emotion_predictions": [
    {
      "emotion": {
        "anger": 0.6145385088342609,
        "disgust": 0.016446104781194168,
        "fear": 0.09267639538103885,
        "joy": 0.002527286676040879,
        "sadness": 0.19656945832751013
      },
      "target": "",
      "emotion_mentions": [
        {
          "span": {
            "begin": 0,
            "end": 39,
            "text": "I'm so annoyed that this code won't run"
          },
          "emotion": {
            "anger": 0.6145385088342609,
            "disgust": 0.016446104781194168,
            "fear": 0.09267639538103885,
            "joy": 0.002527286676040879,
            "sadness": 0.19656945832751013
          }
        }
      ]
    }
  ],
  "producer_id": {
    "name": "Ensemble Aggregated Emotion Workflow",
    "version": "0.0.1"
  }
}