DEBBY-YEH/finetuned-laborlaw-bert
07
📘 勞基法條文分類模型(BERT Fine-tuned)
本專案為使用 bert-base-chinese 對台灣勞基法條文進行 多類別分類(Multi-class Classification) 的微調模型。 模型將應用於後續 AI Chatbot 專案 中,輔助分類提問意圖。
🔧 模型資訊
- 預訓練模型:`bert-base-chinese`
- 微調任務:多類別分類(7 類)
- 訓練資料:台灣勞動部法規資料庫(句子層級斷句+人工標註)
- 框架工具:Hugging Face Transformers + PyTorch
- 輸出格式:支援 CLI 模式輸入問句,回傳分類結果與信心分數
📂 分類標籤(共 7 類)
🧪 CLI Demo(使用 inference.py)
python inference.py --text "我想請育嬰留停"<img src="https://huggingface.co/DEBBY-YEH/finetuned-laborlaw-bert/resolve/main/demo_output.png" alt="推論示意圖" width="650"/>
📦 檔案說明
🔗 延伸應用
此分類模型將整合至以下應用專案中:
屆時將整合 Streamlit 前端與 OpenAI 回答機制,提升使用者互動體驗。
📌 TODO
- [ ] 將 demo 整合至 Streamlit 網頁應用
- [ ] 設計 API 端點供 chatbot 即時調用
- [ ] 加入更多條文與分類(如育嬰假、遠距工作等新興議題)
🙏 若本模型對您有幫助,歡迎給予 ⭐️ star 或連結您的應用專案。
