Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,11 +54,11 @@ if st.button("Предсказать цену"):
|
|
| 54 |
st.write(f"Прогнозируемая цена авто: {predicted_price} руб.")
|
| 55 |
|
| 56 |
|
| 57 |
-
#
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
#
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
|
|
|
| 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()
|