Spaces:
Build error
Build error
Upload app.py
Browse files
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)
|