mithun-ai commited on
Commit
01440cb
·
verified ·
1 Parent(s): 734a2af

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -0
README.md CHANGED
@@ -1,3 +1,56 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ tags:
4
+ - healthcare
5
+ - machine-learning
6
+ - lightgbm
7
+ - stroke-prediction
8
+ - binary-classification
9
  ---
10
+
11
+ # Stroke Risk Predictor - LightGBM
12
+
13
+ ## Overview
14
+
15
+ This model predicts stroke risk using healthcare-related patient features.
16
+
17
+ ## Model Type
18
+
19
+ LightGBM Classifier
20
+
21
+ ## Inputs
22
+
23
+ - Age
24
+ - Hypertension
25
+ - Heart Disease
26
+ - Average Glucose Level
27
+ - BMI
28
+ - Smoking Status
29
+ - Other health indicators
30
+
31
+ ## Output
32
+
33
+ - 0 → No Stroke Risk
34
+ - 1 → Stroke Risk
35
+
36
+ ## Performance
37
+
38
+ Accuracy: 91.10%
39
+
40
+ Recall: 28%
41
+
42
+ ROC-AUC: 78.74%
43
+
44
+ ## Usage
45
+
46
+ ```python
47
+ import joblib
48
+
49
+ model = joblib.load("lightgbm_model.joblib")
50
+
51
+ prediction = model.predict(data)
52
+ ```
53
+
54
+ ## Author
55
+
56
+ Mithun