metadata
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]