🚀 DeBERTa 情感預測器
DeBERTa 情感預測器是一個基於微調後的 DeBERTa 模型,用於進行日語文本情感預測的 Python 包。它提供了 8 種情感(喜悅、悲傷、期待、驚訝、憤怒、恐懼、厭惡、信任)的預測模型,能夠輕鬆獲取每個文本的情感預測標籤和肯定類別的置信度。
🚀 快速開始
DeBERTa 情感預測器可以幫助你快速對日語文本進行情感分析,下面將為你介紹如何安裝和使用它。
📦 安裝指南
你可以使用 pip 來安裝本包:
pip install deberta-emotion-predictor
此外,運行該包還需要安裝 torch
、transformers
和 pandas
:
pip install torch
pip install transformers
pip install pandas
⚠️ 重要提示
若要使用 GPU,需要安裝 NVIDIA GPU 驅動程序等,請參考其他資料進行安裝。
💻 使用示例
基礎用法
from deberta_emotion_predictor import DeBERTaEmotionPredictor
predictor = DeBERTaEmotionPredictor()
result = predictor.predict_emotions("今日はとても嬉しい!")
predictor.show_emotions(result)
高級用法
你也可以輸入一個數據框進行批量預測:
import pandas as pd
from deberta_emotion_predictor import DeBERTaEmotionPredictor
predictor = DeBERTaEmotionPredictor()
sample_texts = [
"そうだ 京都、行こう。",
"がんばるひとの、がんばらない時間。",
"わたしらしくをあたらしく",
"ピースはここにある。",
"結婚しなくても幸せになれるこの時代に、私は、あなたと結婚したいのです。",
"これからの地球のために一肌、脫ぎました。",
"自分は、きっと想像以上だ。",
"ハローしあわせ。",
"日本を、1枚で。"
]
res_df = predictor.predict_emotions(sample_texts)
predictor.show_emotions(res_df)
⚠️ 重要提示
由於需要從 Hugging-face 下載 8 種 DeBERTa 模型,首次啟動會非常耗時,後續運行會加快。
文本傳遞方式(列表)
sample_texts = [
"そうだ 京都、行こう。",
"がんばるひとの、がんばらない時間。"
]
result_df = predictor.predict_emotions(sample_texts)
predictor.show_emotions(result_df)
單一文本的情況
result_single = predictor.predict_emotions("新しい朝が來た。")
print(result_single)
輸出的數據框
輸出的數據框包含表示每種情感是否存在的 8 列,以及每種情感的概率值:
print(result_df)
✨ 主要特性
- 8 種情感預測:使用針對每種情感微調的模型進行文本情感預測。
- 靈活的輸入形式:接受單一文本、文本列表或 pandas Series 作為輸入,並以 DataFrame 形式返回結果。
- 高效推理:為了減少 GPU 內存使用,僅在需要時將模型加載到 GPU。
📚 詳細文檔
目錄結構
deberta_emotion_predictor/
├── README.md # 此說明文件
├── deberta_emotion_predictor.py # DeBERTaEmotionPredictor 類的實現
│ └── tokenizer_DeBERTa_v3_large/ # 分詞器
├── setup.py
├── pyproject.toml
├── README.md
├── LICENSE
└── usage.py
必要環境
- Python 3.6 及以上
- PyTorch
- transformers
- pandas
📄 許可證
本項目採用知識共享署名 - 相同方式共享 4.0 國際許可協議(CC BY - SA 4.0)。
版權所有 (c) 2025 Yoichi Takenaka
本作品採用知識共享署名 - 相同方式共享 4.0 國際許可協議進行許可。要查看此許可協議的副本,請訪問 https://creativecommons.org/licenses/by - sa/4.0/。
本項目基於以下內容:
- DeBERTa (https://huggingface.co/microsoft/deberta - v3 - large),採用 MIT 許可協議。
- DeBERTa 日語模型 (https://huggingface.co/globis - university/deberta - v3 - japanese - large),採用 CC BY - SA 4.0 許可協議。
任何修改或派生作品也必須根據相同的 CC BY - SA 4.0 許可協議進行分發。