---
license: mit
language:
  - ko
tags:
  - disease-classification
  - symptom-analysis
  - medical-nlp
  - korean
pipeline_tag: text-classification
library_name: pytorch
---

# 🧠 Disease Prediction Model (Korean)
> 의료 자연어 처리 기반 한국어 질병 예측 모델  
> Korean disease prediction model based on symptom description and clinical NLP.

---

## 📌 Model Overview | 모델 개요

이 모델은 환자의 증상 설명(텍스트)을 입력받아, 해당 증상에 가장 관련된 질병들을 예측합니다.  
Python + PyTorch로 학습되었으며, 한국어 데이터셋을 기반으로 구성되었습니다.

This model takes Korean symptom descriptions as input and outputs the most likely diseases.  
It is trained on symptom-disease datasets using PyTorch.

---

## 📁 Files | 포함된 파일

- `disease_classifier_epoch10.pt`: PyTorch 기반의 모델 가중치 (약 395MB)

---

## 🛠 Intended Use | 사용 목적

- **의료 챗봇**, 자가진단 보조 시스템
- **한국어 임상 텍스트** 기반 질병 분류 실험

---

## 🚫 Limitations | 한계 사항

- 실제 의학적 진단을 대체하지 않습니다.
- 제한된 증상-질병 매핑 데이터셋에 기반함
- 일부 희귀 질환은 예측 성능이 낮을 수 있습니다.

---

## 📬 Contact | 문의

본 모델은 충북대학교 졸업작품 프로젝트의 일환으로 개발되었습니다.  
개발팀: CBNU Development Team [tuktak]

---