Dbert Sentiment

🚀 基於BERT的情感分類項目

本項目利用Hugging Face的transformers庫，藉助預訓練的BERT模型實現情感分類功能，可對輸入文本進行情感極性的判斷。

🚀 快速開始

以下是使用預訓練的BERT模型進行情感分類的代碼示例：

from transformers import BertForSequenceClassification, BertTokenizer, TextClassificationPipeline
model = BertForSequenceClassification.from_pretrained("deeq/dbert-sentiment")
tokenizer = BertTokenizer.from_pretrained("deeq/dbert")
nlp = TextClassificationPipeline(model=model, tokenizer=tokenizer)

# 測試積極情感文本
print(nlp("좋아요"))
# 測試中性或消極情感文本
print(nlp("글쎄요"))

以上代碼首先從transformers庫中導入必要的類，然後加載預訓練的模型和分詞器，接著創建一個文本分類管道，最後對示例文本進行情感分類並打印結果。