25 lines
1.1 KiB
Markdown
25 lines
1.1 KiB
Markdown
---
|
||
base_model: google/gemma-2-9b-it
|
||
library_name: transformers
|
||
license: gemma
|
||
language: [en, vi]
|
||
tags: [cbt, mental-health, counseling, lora, structured-output]
|
||
pipeline_tag: text-generation
|
||
---
|
||
|
||
# cbt-gemma2-9b
|
||
|
||
Fine-tune (LoRA→merged) của **google/gemma-2-9b-it** cho trợ lý tham vấn theo liệu pháp CBT.
|
||
Model sinh **JSON có cấu trúc** gồm: `risk_level · technique · rationale · plan · response · safety_action`.
|
||
|
||
## Huấn luyện
|
||
- Dữ liệu: 26.000 mẫu hội thoại CBT (định dạng `messages`, output JSON 6 trường).
|
||
- LoRA r=32/α=64, lr=5e-05, max_len=1536, bf16.
|
||
- best eval_loss = 0.46450522541999817 (epoch 2.0).
|
||
|
||
## ⚠️ Giới hạn & an toàn
|
||
Đây là **công cụ hỗ trợ học thuật**, KHÔNG thay thế chuyên gia sức khỏe tâm thần,
|
||
KHÔNG chẩn đoán, KHÔNG tư vấn thuốc. Khi có dấu hiệu khủng hoảng/tự hại, model
|
||
được huấn luyện để ưu tiên an toàn và khuyến nghị liên hệ chuyên gia/dịch vụ khẩn cấp.
|
||
Cần có chuyên gia rà soát (human-in-the-loop) trước khi dùng với người thật.
|