--- language: zh tags: - chinese - emotion - classification license: mit --- # Chinese Emotion Classification Model 这是一个中文情感分类模型,可以将文本分类为三种情感: - 快乐 - 愤怒 - 悲伤 ## 使用方法 ```python from transformers import pipeline # 创建分类器 classifier = pipeline("text-classification", model="WJL110/emotion-classifier") # 标签映射 label_map = { "LABEL_0": "快乐", "LABEL_1": "愤怒", "LABEL_2": "悲伤" } # 测试文本 test_texts = [ "今天真是太开心了!", "这件事让我很生气。", "听到这个消息很难过。" ] print("=== 情感分析测试 ===") for text in test_texts: result = classifier(text)[0] # 获取第一个(也是唯一的)结果 emotion = label_map[result['label']] confidence = result['score'] print(f"\n输入文本: {text}") print(f"预测情感: {emotion}") print(f"置信度: {confidence:.2f}") ```