TitanicApp / app1.py
Shivam2396's picture
Upload app1.py
ce29796
raw
history blame
No virus
2.1 kB
import pandas as pd
import time
import streamlit as st
import plotly.express as px
from pycaret.classification import *
import streamlit as st
import pandas as pd
model_gr = load_model('deploy_gboost')
model_rf=load_model('deploy_rand_for')
model_lr=load_model('deploy_log_reg')
def predict(model, input_df):
predictions_df = predict_model(estimator=model, data=input_df)
predictions = predictions_df['Label'][0]
return predictions
def app():
from PIL import Image
st.title('Streamlit Example')
st.write("""
# Explore different classifier
""")
st.write("Titanic Dataset")
classifier_name = st.sidebar.selectbox(
'Select classifier',
('Gradient Boost', 'Random Forest', 'Logistic Regression')
)
st.title("Titanic Prediction App")
Age = st.number_input('Age', min_value=1, max_value=100, value=25)
Sex = st.selectbox('Sex', ['male', 'female'])
Pclass= st.number_input('P Class', 1,3)
SibSp= st.multiselect('Number of Siblings And Spouse',[0,1,2,3,4,5,8])
Parch= st.multiselect('Parch',[0,1,2,3,4,5,6])
Fare= st.slider('Fare', 0,600)
Embarked = st.selectbox('Embarked', ['S', 'C', 'Q'])
output=""
input_dict = {'Age' : Age, 'Sex' : Sex, 'Pclass':Pclass,'SibSp':SibSp,'Parch':Parch,'Fare':Fare,'Embarked':Embarked}
input_df = pd.DataFrame([input_dict])
st.dataframe(input_df)
if st.button("Predict"):
if classifier_name=='Gradient Boost':
output = predict(model=model_gr, input_df=input_df)
output = '$' + str(output)
st.success('The output is {}'.format(output))
elif classifier_name=='Random Forest':
output = predict(model=model_rf, input_df=input_df)
output = '$' + str(output)
st.success('The output is {}'.format(output))
else:
output = predict(model=model_lr, input_df=input_df)
output = '$' + str(output)
st.success('The output is {}'.format(output))