C2 Roberta Base Finetuned Dianping Chinese

🚀 liam168/c2-roberta-base-finetuned-dianping-chinese

這是一個使用中文對話情緒語料訓練的模型，可進行樂觀和悲觀的二分類，能有效識別中文文本中的情緒傾向。

🚀 快速開始

此模型可用於對中文文本進行情緒分類，判斷其表達的是樂觀還是悲觀情緒。

✨ 主要特性

• 適用語言：中文。
• 模型類型：基於BertForSequenceClassification的語言模型。
• 模型大小：410M。

💻 使用示例

基礎用法

>>> from transformers import AutoModelForSequenceClassification , AutoTokenizer, pipeline

>>> model_name = "liam168/c2-roberta-base-finetuned-dianping-chinese"
>>> class_num = 2
>>> ts_texts = ["我喜歡下雨。", "我討厭他."]
>>> model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=class_num)
>>> tokenizer = AutoTokenizer.from_pretrained(model_name)
>>> classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
>>> classifier(ts_texts[0])
>>> classifier(ts_texts[1])

[{'label': 'positive', 'score': 0.9973447918891907}]
[{'label': 'negative', 'score': 0.9972558617591858}]