Almas-123 commited on
Commit
a2c46e4
·
verified ·
1 Parent(s): 8e6984d

Create ml_model.py

Browse files
Files changed (1) hide show
  1. ml_model.py +6 -0
ml_model.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+
3
+ def predict_common_diseases(filepath):
4
+ df = pd.read_csv(filepath)
5
+ disease_counts = df['disease'].value_counts().head(5)
6
+ return list(disease_counts.index)