Shivam2396 commited on
Commit
e07e326
1 Parent(s): a606436

Upload app2.py

Browse files
Files changed (1) hide show
  1. app2.py +107 -0
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))