1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning with Tree-Based Models in Python
Connected

cvičení

Vyhodnocení regresního stromu

V tomto cvičení vyhodnotíš výkon modelu dt na testovací sadě pomocí metriky RMSE (Root Mean Squared Error). RMSE měří, o kolik se predikce modelu průměrně liší od skutečných hodnot. Vypočítáš ji jako odmocninu střední kvadratické chyby (MSE) modelu.

V tvém pracovním prostoru jsou k dispozici matice příznaků X_test, pole y_test a rozhodovací strom dt, který jsi natrénoval/a v předchozím cvičení.

Pokyny

100 XP
  • Importuj funkci mean_squared_error jako MSE z modulu sklearn.metrics.
  • Predikuj štítky testovací sady a výsledek ulož do proměnné y_pred.
  • Vypočítej MSE na testovací sadě pomocí funkce MSE a výsledek ulož do mse_dt.
  • Vypočítej RMSE na testovací sadě a ulož ji do rmse_dt.