Medical Chatbot Model

This model is a fine-tuned version of KLUE BERT base for medical symptom classification.

Model description

The model uses KLUE BERT base architecture fine-tuned on a custom medical dataset. It performs classification for various diseases and provides treatment recommendations.

Input Format

{
    "text": "๋‘ํ†ต์ด ์‹ฌํ•˜๊ณ  ๋ฉ”์Šค๊บผ์›Œ์š”"
}

Output Format

{
    "label": "ํŽธ๋‘ํ†ต",
    "score": 0.95
}

Training procedure

  • Epochs: 5
  • Batch size: 8
  • Learning rate: 2e-5
  • Maximum sequence length: 128

Limitations and bias

This model should be used for reference only and not as a substitute for professional medical advice.

Downloads last month
23
Safetensors
Model size
111M params
Tensor type
F32
ยท
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.