CoolFace
Modelpublic

DEBBY-YEH/finetuned-laborlaw-bert

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes7downloads
Model Card

📘 勞基法條文分類模型(BERT Fine-tuned)

本專案為使用 bert-base-chinese 對台灣勞基法條文進行 多類別分類(Multi-class Classification) 的微調模型。 模型將應用於後續 AI Chatbot 專案 中,輔助分類提問意圖。


🔧 模型資訊

  • —預訓練模型:`bert-base-chinese`
  • —微調任務:多類別分類(7 類)
  • —訓練資料:台灣勞動部法規資料庫(句子層級斷句+人工標註)
  • —框架工具:Hugging Face Transformers + PyTorch
  • —輸出格式:支援 CLI 模式輸入問句,回傳分類結果與信心分數

📂 分類標籤(共 7 類)

類別 ID分類標籤
0假別
1其他
2契約與聘僱關係
3工時
4終止與解僱
5職場安全與性別平等
6薪資

🧪 CLI Demo(使用 inference.py)

bash
python inference.py --text "我想請育嬰留停"

<img src="https://huggingface.co/DEBBY-YEH/finetuned-laborlaw-bert/resolve/main/demo_output.png" alt="推論示意圖" width="650"/>


📦 檔案說明

檔案名稱說明
model.safetensorsFine-tuned 模型主體
tokenizer_config.json / vocab.txt / tokenizer.json中文 BERT tokenizer 相關設定
label2id.json類別對應表
inference.pyCLI 模型推論腳本
demo_output.png預測示意圖

🔗 延伸應用

此分類模型將整合至以下應用專案中:

屆時將整合 Streamlit 前端與 OpenAI 回答機制,提升使用者互動體驗。


📌 TODO

  • —[ ] 將 demo 整合至 Streamlit 網頁應用
  • —[ ] 設計 API 端點供 chatbot 即時調用
  • —[ ] 加入更多條文與分類(如育嬰假、遠距工作等新興議題)

🙏 若本模型對您有幫助,歡迎給予 ⭐️ star 或連結您的應用專案。