peggy30 commited on
Commit
90be8db
·
1 Parent(s): 927f088
Files changed (1) hide show
  1. pages/ICE_and_PDP.py +3 -3
pages/ICE_and_PDP.py CHANGED
@@ -55,13 +55,13 @@ def main():
55
  selected_feature = st.sidebar.selectbox("Select a feature for PDP/ICE analysis:", ("Age", "Workclass", "Education-Num", "Marital Status", "Occupation",
56
  "Relationship", "Race", "Sex", "Capital Gain", "Capital Loss", "Hours per week", "Country"),)
57
  print(f"selected feature is {selected_feature}")
58
- kind = st.sidebar.selectbox("Select plot type:", ("average", "both", "individual"),)
59
 
60
- st.title("ICE (Individual Conditional Expectation) and PDP (Partial Dependence Plot)")
61
  st.write(prompt_params.ICE_INTRODUCTION)
62
  # Explain the selected sample
63
  if st.button("Explain Sample"):
64
- explain_example(selected_feature, kind)
65
 
66
 
67
  if __name__ == '__main__':
 
55
  selected_feature = st.sidebar.selectbox("Select a feature for PDP/ICE analysis:", ("Age", "Workclass", "Education-Num", "Marital Status", "Occupation",
56
  "Relationship", "Race", "Sex", "Capital Gain", "Capital Loss", "Hours per week", "Country"),)
57
  print(f"selected feature is {selected_feature}")
58
+ kind = st.sidebar.selectbox("Select plot type:", ("individual", "average", "both"),)
59
 
60
+ st.title("ICE and PDP")
61
  st.write(prompt_params.ICE_INTRODUCTION)
62
  # Explain the selected sample
63
  if st.button("Explain Sample"):
64
+ explain_example("Age", kind)
65
 
66
 
67
  if __name__ == '__main__':