Emotion English

🚀 20クラスからの感情分類

このモデルは、20種類の感情ラベルを用いてテキストの感情分類を行うことができます。特定のテキストから感情を識別し、適切なラベルを付けることが可能です。

🚀 クイックスタート

このモデルを使用して、与えられたテキストの感情ラベルを取得する方法を説明します。

💻 使用例

基本的な使用法

from transformers import AutoModelForSequenceClassification, pipeline

model_name = 'jitesh/emotion-english'
model = AutoModelForSequenceClassification.from_pretrained(model_name)
classifier = pipeline("text-classification", model=model, tokenizer=model_name)

text = "I can't wait any longer "

prediction = classifier(text)
print(prediction[0], text)

実行結果

上記のコードを実行すると、以下のような出力が得られます。

{'label': 'impatient', 'score': 0.924211859703064} I can't wait any longer 

📚 ドキュメント

20の感情ラベル

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。