Title model update

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 import random
 import streamlit as st
 
-model = load("model.joblib")
+model = load("model_title_update.joblib")
 
 session = st.session_state
 
@@ -26,6 +26,11 @@ age = st.number_input('Age', min_value=1, max_value=80)
 
 sex = st.radio("Sex", ("Male", "Female"))
 
+if sex == "Male":
+    title = st.radio("Title", ("Doctor", "Master", "Reverend"))
+else:
+    title = st.radio("Title", ("Doctor", "Master", "Miss", "Mrs", "Reverend"))
+
 embark_port = st.radio(
     "Port of Embarkation",
     ('Cherbourg', 'Queenstown', 'Southampton'))
@@ -34,9 +39,9 @@ p_class = st.radio(
     "Passenger Class",
     ('Class 1', 'Class 2', 'Class 3'))
 
-siblings = st.number_input('Number of Siblings/Spouses Aboard', min_value=0, max_value=5)
+siblings = st.number_input('Number of Spouses Aboard', min_value=0, max_value=5)
 
-parents = st.number_input("Number of Parents/Children Aboard", min_value=0, max_value=6)
+parents = st.number_input("Number of Children Aboard", min_value=0, max_value=6)
 
 
 passenger_data = {
@@ -50,6 +55,13 @@ passenger_data = {
     'Embarked_S': 1 if embark_port[0] == "S" else 0,
     'SibSp': siblings,
     'Parch': parents,
+    'Mr': 1 if title == "Mr" else 0,
+    'Miss': 1 if title == "Miss" else 0,
+    'Mrs': 1 if title == "Mrs" else 0,
+    "Master": 1 if title == "Master" else 0,
+    'Dr': 1 if title == "Doctor" else 0,
+    "Rev": 1 if title == "Reverend" else 0,
+    'Other': 0,
 }
 
 if st.button("Calculate my chances of survival"):

requirements.txt

@@ -1,3 +1,3 @@
 joblib
 pandas
-scikit-learn==1.0.2
+scikit-learn==1.2.2