bbhagat1 commited on
Commit
92de636
1 Parent(s): 4ad1ee9

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import pickle
3
+ import numpy as np
4
+ from sklearn import tree
5
+ # Load the Random Forest CLassifier model
6
+ filename = 'model.pkl'
7
+ loaded_model = pickle.load(open(filename, 'rb'))
8
+ print(loaded_model)
9
+
10
+ input = [0.0,0.0,1.0,26.0,1.0,0.0,0.0,0.0,0.0,1.0,0.0,1.0,0.0,3.0,0.0,15.0,0.0,0.0,7.0,5.0,7.0]
11
+ manualInput =[
12
+ [1.0,1.0,1.0,37.0,1.0,1.0,1.0,0.0,0.0,1.0,0.0,1.0,0.0,5.0,0.0,0.0,1.0,1.0,10.0,6.0,5.0]
13
+ #,[1.0,1.0,1.0,28.0,1.0,0.0,1.0,0.0,0.0,1.0,0.0,1.0,0.0,4.0,0.0,0.0,0.0,1.0,12.0,2.0,4.0]
14
+ #,[1.0,1.0,1.0,27.0,1.0,0.0,0.0,0.0,1.0,1.0,0.0,1.0,0.0,4.0,20.0,20.0,1.0,0.0,8.0,4.0,7.0]
15
+ ]
16
+ print(input, manualInput)
17
+ col=["HighBP","HighChol","CholCheck","BMI","Smoker","Stroke","HeartDiseaseorAttack","PhysActivity","Fruits"
18
+ ,"Veggies","HvyAlcoholConsump","AnyHealthcare","NoDocbcCost","GenHlth","MentHlth","PhysHlth","DiffWalk"
19
+ ,"Sex","Age","Education","Income"]
20
+ ddf = pd.DataFrame(manualInput, columns=col)
21
+ print(ddf)
22
+
23
+ result = loaded_model.predict(ddf)
24
+ print(result)