Update README.md
Browse files
README.md
CHANGED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
language:
|
4 |
+
- ko
|
5 |
+
tags:
|
6 |
+
- disease-classification
|
7 |
+
- symptom-analysis
|
8 |
+
- medical-nlp
|
9 |
+
- korean
|
10 |
+
pipeline_tag: text-classification
|
11 |
+
library_name: pytorch
|
12 |
+
---
|
13 |
+
|
14 |
+
# 🧠 Disease Prediction Model (Korean)
|
15 |
+
> 의료 자연어 처리 기반 한국어 질병 예측 모델
|
16 |
+
> Korean disease prediction model based on symptom description and clinical NLP.
|
17 |
+
|
18 |
+
---
|
19 |
+
|
20 |
+
## 📌 Model Overview | 모델 개요
|
21 |
+
|
22 |
+
이 모델은 환자의 증상 설명(텍스트)을 입력받아, 해당 증상에 가장 관련된 질병들을 예측합니다.
|
23 |
+
Python + PyTorch로 학습되었으며, 한국어 데이터셋을 기반으로 구성되었습니다.
|
24 |
+
|
25 |
+
This model takes Korean symptom descriptions as input and outputs the most likely diseases.
|
26 |
+
It is trained on symptom-disease datasets using PyTorch.
|
27 |
+
|
28 |
+
---
|
29 |
+
|
30 |
+
## 📁 Files | 포함된 파일
|
31 |
+
|
32 |
+
- `disease_classifier_epoch10.pt`: PyTorch 기반의 모델 가중치 (약 395MB)
|
33 |
+
|
34 |
+
---
|
35 |
+
|
36 |
+
## 🛠 Intended Use | 사용 목적
|
37 |
+
|
38 |
+
- **의료 챗봇**, 자가진단 보조 시스템
|
39 |
+
- **한국어 임상 텍스트** 기반 질병 분류 실험
|
40 |
+
|
41 |
+
---
|
42 |
+
|
43 |
+
## 🚫 Limitations | 한계 사항
|
44 |
+
|
45 |
+
- 실제 의학적 진단을 대체하지 않습니다.
|
46 |
+
- 제한된 증상-질병 매핑 데이터셋에 기반함
|
47 |
+
- 일부 희귀 질환은 예측 성능이 낮을 수 있습니다.
|
48 |
+
|
49 |
+
---
|
50 |
+
|
51 |
+
## 📬 Contact | 문의
|
52 |
+
|
53 |
+
본 모델은 충북대학교 졸업작품 프로젝트의 일환으로 개발되었습니다.
|
54 |
+
개발팀: CBNU Development Team [tuktak]
|
55 |
+
|
56 |
+
---
|