Solar-Iz commited on
Commit
7064dcf
·
1 Parent(s): 5eb1934

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -54,11 +54,11 @@ if st.button("Предсказать цену"):
54
  st.write(f"Прогнозируемая цена авто: {predicted_price} руб.")
55
 
56
 
57
- # # Расчет важности фичей с использованием SHAP
58
- # explainer = shap.TreeExplainer(final_model)
59
- # shap_values = explainer.shap_values(input_data)
60
-
61
- # # Отображение SHAP force plot
62
- # st.write("SHAP Force Plot:")
63
- # shap.force_plot(explainer.expected_value, shap_values, input_data, matplotlib=True, show=False)
64
- # st.pyplot()
 
54
  st.write(f"Прогнозируемая цена авто: {predicted_price} руб.")
55
 
56
 
57
+ # Расчет важности фичей с использованием SHAP
58
+ explainer = shap.TreeExplainer(final_model)
59
+ shap_values = explainer.shap_values(input_data)
60
+
61
+ # Отображение SHAP force plot
62
+ st.write("SHAP Force Plot:")
63
+ shap.force_plot(explainer.expected_value, shap_values, input_data, matplotlib=True, show=False)
64
+ st.pyplot()