jtdatastoryteller/bert-base-chinese-issue-finetuned-pchome

中文自定義分類模型

本模型基於 bert-base-chinese，針對中文評論文本進行六種分類的 Fine-tuning。

模型用途

輸入一段 App 使用問題，模型會自動將其分類至以下六個類別之一：

使用方式

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="你的帳號名稱/bert-base-chinese-issue-finetuned-pchome"
)

result = classifier("我的訂單付款後一直顯示處理中，請問正常嗎？")
print(result)
# 輸出：[{'label': '結帳付款', 'score': 0.94}]

訓練資料

• —資料集：Pchome App 評論情緒分類標註資料
• —樣本數：約 1,000 筆

訓練設定

• —基礎模型：bert-base-chinese
• —學習率：2e-5
• —批次大小：32
• —最大序列長度：128
• —損失函數：Focal Loss（gamma=2）
• —不平衡處理：Class Weight

注意事項

本模型以小規模資料集訓練，建議用於同領域的分析場景。 若應用於其他領域，建議重新進行 Fine-tuning。