中文自定義分類模型

本模型基於 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。

Downloads last month
2
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support