updating

app.py

@@ -7,7 +7,6 @@ print(st.__version__)
 print(pickle.format_version)
 
 st.title("EMPLOYEE ATTRITION RISK PREDICTION")
-
 st.write('Created by Hasbi Thaufik Oktodila')
 
 # Step 1 - import saved model
@@ -16,19 +15,11 @@ model = pickle.load(open("model.pkl", "rb"))
 st.write('Insert feature to predict')
 
 # Step 2 - prepare input data for user
-
-# jobrole = st.selectbox('Job Role', df['jobrole'].unique())
-# maritalstatus = st.selectbox('Marital Status', df['maritalstatus'].unique())
-# overtime = st.selectbox('Frequently Overtime? (Yes = 1, No = 0)', df['overtime'].unique())
-# businesstravel = st.selectbox('Frequently Have Business Travel? (No Travel = 1, Rarely = 2, Frequently = 3', df['businesstravel'].unique())
-# gender = st.selectbox('Gender (Male = 1, Female = 2)', df['gender'].unique())
-# hourlyrate = st.number_input('Hourly Rate', df['degree_t'].unique())
-
 jobrole = st.selectbox(label='Job Role', options=['Sales Executive', 'Laboratory Technician', 'Research Scientist', 'Manufacturing Director', 'Healthcare Representative',
                                                   'Manager', 'Research Director', 'Sales Representative', 'Human Resources'])
 maritalstatus = st.selectbox(label='Marital Status', options=['Single', 'Married', 'Divorced'])
 overtime = st.selectbox(label='Frequently Overtime? (Yes = 1, No = 0)', options=[0, 1])
-businesstravel = st.selectbox(label='Frequently Have Business Travel? (Yes = 1, No = 0)', options=[1, 2, 3])
+businesstravel = st.selectbox(label='Frequently Have Business Travel? (1 = Non Travel, 2 = Rarely, 3 = Frequently)', options=[1, 2, 3])
 gender = st.selectbox(label='Gender (Male = 1, Female = 2)', options=[1,2])
 hourlyrate = st.slider(label='Your Hourly Rate', min_value=30, max_value=100, step=1)
 dailyrate = st.slider(label='Your Daily Rate', min_value=100, max_value=2000)