Solar-Iz commited on
Commit
d76225e
·
1 Parent(s): b44b271

Update pages/metrics.py

Browse files
Files changed (1) hide show
  1. pages/metrics.py +7 -2
pages/metrics.py CHANGED
@@ -32,7 +32,7 @@ st.title("Метрики модели")
32
  # Построение гистограммы цен
33
  plt.figure(figsize=(10, 6))
34
  plt.hist(y_test_trimmed_loaded, bins=50, color='red', alpha=0.7, label='Цены на тестовой выборке')
35
- plt.title('Распределение цен на автомобили')
36
  plt.xlabel('Цена')
37
  plt.ylabel('Количество автомобилей')
38
  plt.legend()
@@ -41,15 +41,20 @@ st.pyplot()
41
  # Предсказание на тестовых данных
42
  y_pred_final = final_model.predict(X_test_trimmed_loaded)
43
 
44
- # Вычисляем MAE, RMSE и R^2
 
 
 
45
  mae_final = mean_absolute_error(y_test_trimmed_loaded, y_pred_final)
46
  rmse_final = mean_squared_error(y_test_trimmed_loaded, y_pred_final, squared=False) # RMSE
47
  r2_final = r2_score(y_test_trimmed_loaded, y_pred_final)
 
48
 
49
  # Вывод метрик
50
  st.write(f"Final Mean Absolute Error (MAE): {mae_final}")
51
  st.write(f"Final Root Mean Squared Error (RMSE): {rmse_final}")
52
  st.write(f"Final R^2 Score: {r2_final}")
 
53
 
54
  # Возвращение на первую страницу
55
  st.markdown("---")
 
32
  # Построение гистограммы цен
33
  plt.figure(figsize=(10, 6))
34
  plt.hist(y_test_trimmed_loaded, bins=50, color='red', alpha=0.7, label='Цены на тестовой выборке')
35
+ plt.title('Распределение цен на автомобили после исключения выбросов после правого перцентиля 0.95')
36
  plt.xlabel('Цена')
37
  plt.ylabel('Количество автомобилей')
38
  plt.legend()
 
41
  # Предсказание на тестовых данных
42
  y_pred_final = final_model.predict(X_test_trimmed_loaded)
43
 
44
+ def mean_absolute_percentage_error(y_true, y_pred):
45
+ return 100 * np.mean(np.abs((y_true - y_pred) / y_true))
46
+
47
+ # Вычисляем MAE, RMSE и R^2 и MAPE
48
  mae_final = mean_absolute_error(y_test_trimmed_loaded, y_pred_final)
49
  rmse_final = mean_squared_error(y_test_trimmed_loaded, y_pred_final, squared=False) # RMSE
50
  r2_final = r2_score(y_test_trimmed_loaded, y_pred_final)
51
+ mape_final = mean_absolute_percentage_error(y_test_trimmed_loaded, y_pred_final) # Добавляем MAPE
52
 
53
  # Вывод метрик
54
  st.write(f"Final Mean Absolute Error (MAE): {mae_final}")
55
  st.write(f"Final Root Mean Squared Error (RMSE): {rmse_final}")
56
  st.write(f"Final R^2 Score: {r2_final}")
57
+ st.write(f"Final Mean Absolute Percentage Error (MAPE): {mape_final:.2f}%")
58
 
59
  # Возвращение на первую страницу
60
  st.markdown("---")