Spaces:
Build error
Build error
Shivam2396
commited on
Commit
•
e07e326
1
Parent(s):
a606436
Upload app2.py
Browse files
app2.py
ADDED
@@ -0,0 +1,107 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import pandas as pd
|
2 |
+
import streamlit as st
|
3 |
+
|
4 |
+
|
5 |
+
def app():
|
6 |
+
import joblib
|
7 |
+
st.title('SKLEARN')
|
8 |
+
st.write('Welcome to app2 sklearn')
|
9 |
+
st.title('Streamlit Example')
|
10 |
+
st.write("""
|
11 |
+
# Explore different classifier
|
12 |
+
""")
|
13 |
+
st.write("Titanic Dataset")
|
14 |
+
|
15 |
+
Pclass = st.number_input('P Class', 1, 3)
|
16 |
+
Sex = st.selectbox('Sex', ['male', 'female'])
|
17 |
+
Age = st.number_input('Age', min_value=1, max_value=100, value=25)
|
18 |
+
Fare = st.slider('Fare', 0, 600)
|
19 |
+
Cabin = st.selectbox('Cabin', [0, 0.4, 0.8, 1.2, 1.6, 2, 2.4, 2.8])
|
20 |
+
Embarked = st.selectbox('Embarked', ['S', 'C', 'Q'])
|
21 |
+
#Title = st.selectbox('Title', ['Mr', 'Ms', 'Mrs', 'Master', 'Others'])
|
22 |
+
#SibSp= st.selectbox('Number of Siblings And Spouse',[0,1,2,3,4,5,8])
|
23 |
+
#Parch= st.selectbox('Parch',[0,1,2,3,4,5,6])
|
24 |
+
#FamilySize = int(SibSp + Parch + 1)
|
25 |
+
FamilySize = st.slider('Family size', 1, 11)
|
26 |
+
|
27 |
+
if Sex == "male":
|
28 |
+
Title = st.selectbox('Title', ['Mr', 'Master', 'Others'])
|
29 |
+
else:
|
30 |
+
Title = st.selectbox('Title', ['Ms', 'Mrs', 'Others'])
|
31 |
+
|
32 |
+
input_dict = {
|
33 |
+
'Pclass': Pclass,
|
34 |
+
'Sex': Sex,
|
35 |
+
'Age': Age,
|
36 |
+
'Fare': Fare,
|
37 |
+
'Cabin': Cabin,
|
38 |
+
'Embarked': Embarked,
|
39 |
+
'Title': Title,
|
40 |
+
'FamilySize': FamilySize}
|
41 |
+
input_df = pd.DataFrame([input_dict])
|
42 |
+
|
43 |
+
dic_sex = {"male": 0, "female": 1}
|
44 |
+
input_df["Sex"] = input_df["Sex"].map(dic_sex)
|
45 |
+
|
46 |
+
|
47 |
+
title_mapping = {'Mr': 0, 'Ms': 1, 'Mrs': 2, 'Master': 3, 'Others': 4}
|
48 |
+
input_df['Title'] = input_df['Title'].map(title_mapping)
|
49 |
+
|
50 |
+
embarked_mapping = {"S": 0, "C": 1, "Q": 2}
|
51 |
+
input_df['Embarked'] = input_df['Embarked'].map(embarked_mapping)
|
52 |
+
|
53 |
+
#cabin_mapping = {"A": 0, "B": 0.4, "C": 0.8, "D": 1.2, "E": 1.6, "F": 2, "G": 2.4, "T": 2.8}
|
54 |
+
#input_df['Cabin'] = input_df['Cabin'].map(cabin_mapping)
|
55 |
+
|
56 |
+
family_mapping = {
|
57 |
+
1: 0,
|
58 |
+
2: 0.4,
|
59 |
+
3: 0.8,
|
60 |
+
4: 1.2,
|
61 |
+
5: 1.6,
|
62 |
+
6: 2,
|
63 |
+
7: 2.4,
|
64 |
+
8: 2.8,
|
65 |
+
9: 3.2,
|
66 |
+
10: 3.6,
|
67 |
+
11: 4}
|
68 |
+
input_df['FamilySize'] = input_df['FamilySize'].map(family_mapping)
|
69 |
+
|
70 |
+
if Fare <= 17:
|
71 |
+
input_df["Fare"] = 0
|
72 |
+
elif (Fare > 17 & Fare <= 30):
|
73 |
+
input_df["Fare"] = 1
|
74 |
+
elif (Fare > 30 & Fare <= 100):
|
75 |
+
input_df["Fare"] = 2
|
76 |
+
elif (Fare > 100):
|
77 |
+
input_df["Fare"] = 3
|
78 |
+
|
79 |
+
if Age <= 16:
|
80 |
+
input_df["Age"] = 0
|
81 |
+
elif (Age > 16 and Age <= 25):
|
82 |
+
input_df["Age"] = 1
|
83 |
+
elif (Age > 25 and Age <= 35):
|
84 |
+
input_df["Age"] = 2
|
85 |
+
elif (Age > 35 and Age <= 45):
|
86 |
+
input_df["Age"] = 3
|
87 |
+
elif (Age > 45):
|
88 |
+
input_df["Age"] = 4
|
89 |
+
|
90 |
+
print(input_df)
|
91 |
+
|
92 |
+
st.dataframe(input_df)
|
93 |
+
|
94 |
+
file_upload = st.file_uploader(
|
95 |
+
"Upload sav file for prediction", type=["sav"])
|
96 |
+
|
97 |
+
if file_upload is not None:
|
98 |
+
|
99 |
+
load_clf = joblib.load(file_upload)
|
100 |
+
output = load_clf.predict(input_df)
|
101 |
+
if output == 0:
|
102 |
+
output = "Not survived"
|
103 |
+
else:
|
104 |
+
output = "Survived"
|
105 |
+
|
106 |
+
if st.button("Predict"):
|
107 |
+
st.success('The output is {} '.format(output))
|