qaisar701shan commited on
Commit
11e0111
·
verified ·
1 Parent(s): 18331ef

Create rules_engine.py

Browse files
Files changed (1) hide show
  1. rules_engine.py +30 -0
rules_engine.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # rules_engine.py
2
+
3
+ import pandas as pd
4
+
5
+ def evaluate_health_risks(df: pd.DataFrame) -> pd.DataFrame:
6
+ """
7
+ Very simple health risk scoring engine.
8
+ Takes input dataframe with health data and adds risk scores.
9
+ """
10
+
11
+ results = []
12
+
13
+ for _, row in df.iterrows():
14
+ risks = {}
15
+
16
+ # Example rules (you can expand / modify later)
17
+ risks["risk_Type 2 Diabetes Risk"] = 0.8 if row["glucose"] > 126 else 0.2
18
+ risks["risk_Hypertension Risk"] = 0.7 if row["systolic_bp"] > 140 else 0.3
19
+ risks["risk_Depression/Mood Concern"] = 0.6 if row["sleep_hours"] < 5 else 0.2
20
+ risks["risk_Migraine Risk"] = 0.5 if row["screen_time"] > 8 else 0.2
21
+ risks["risk_Sleep Apnea Risk"] = 0.7 if row["bmi"] > 30 else 0.3
22
+ risks["risk_Anemia Risk"] = 0.6 if row["hemoglobin"] < 12 else 0.2
23
+
24
+ results.append(risks)
25
+
26
+ # Merge with original dataframe
27
+ risks_df = pd.DataFrame(results)
28
+ final_df = pd.concat([df, risks_df], axis=1)
29
+
30
+ return final_df