🚀 情感分类器：使用句子转换器进行情感检测

本模型使用句子转换器（Sentence Transformer），具体为all-MiniLM-L6-v2架构来对情感进行分类。它在dair-ai/emotion数据集上进行训练，能够识别六种基本情绪：

• 悲伤
• 喜悦
• 爱意
• 愤怒
• 恐惧
• 惊讶

开发者： shhossain: https://github.com/shhossain

🚀 快速开始

本模型使用all-MiniLM-L6-v2架构的句子转换器进行情感分类，在dair-ai/emotion数据集上训练，可识别六种基本情绪。

✨ 主要特性

• 架构：采用all-MiniLM-L6-v2架构。
• 数据集：在dair-ai/emotion数据集上训练。
• 识别能力：能够识别悲伤、喜悦、爱意、愤怒、恐惧和惊讶六种基本情绪。

📦 安装指南

暂未提及安装步骤，可参考transformers库的安装方式来使用此模型。

💻 使用示例

基础用法

from transformers import pipeline

pipe = pipeline("text-classification", model="shhossain/all-MiniLM-L6-v2-sentiment-classifier", trust_remote_code=True)

result = pipe("This product is excellent!")
result

输出：

[{'label': 'sad', 'score': 0.006396006792783737},
 {'label': 'joy', 'score': 0.7897642254829407},
 {'label': 'love', 'score': 0.17318710684776306},
 {'label': 'anger', 'score': 0.008878232911229134},
 {'label': 'fear', 'score': 0.010075093246996403},
 {'label': 'surprise', 'score': 0.011699344962835312}]

📚 详细文档

📄 许可证

许可证与all-MiniLM-L6-v2相同。