rasmodev commited on
Commit
ef5252c
Β·
verified Β·
1 Parent(s): 2985ba9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -38,7 +38,7 @@ st.markdown("πŸ”— **Learn more about employee attrition from [Academy to Innovat
38
  st.markdown("---")
39
 
40
  # Additional Information for Sample Prediction
41
- st.write("πŸ“Š To make a sample prediction, you can refer to the training dataset information available in the sidebar or input the information of the employee whose attrition you want to predict.")
42
  st.write("Please provide the following information to make a prediction:")
43
 
44
  # About Section with Style
@@ -48,14 +48,6 @@ st.sidebar.info(
48
  "It utilizes a CatBoost machine learning model trained on an employee attrition dataset."
49
  )
50
 
51
- # Load The Train Dataset
52
- train_df = pd.read_csv("train_data.csv", index_col=None)
53
-
54
- # Training Dataset Information in the sidebar
55
- st.sidebar.markdown("πŸ“Š **Training Dataset Information:**")
56
- st.sidebar.write("The model is trained on a sepsis dataset. Here's an overview of the dataset:")
57
- st.sidebar.write(train_df.head())
58
-
59
  # Auto-expand sidebar code
60
  st.markdown(
61
  """
 
38
  st.markdown("---")
39
 
40
  # Additional Information for Sample Prediction
41
+ st.write("πŸ“Š To make a prediction, input the information of the employee whose attrition you want to predict.")
42
  st.write("Please provide the following information to make a prediction:")
43
 
44
  # About Section with Style
 
48
  "It utilizes a CatBoost machine learning model trained on an employee attrition dataset."
49
  )
50
 
 
 
 
 
 
 
 
 
51
  # Auto-expand sidebar code
52
  st.markdown(
53
  """