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

cvičení

Vyhodnocení chyby na trénovací množině

Teď vyhodnotíš hodnotu RMSE na trénovací množině, které dosáhlo regresní rozhodovací strom dt vytvořený v předchozím cvičení.

Vedle dt máš v prostředí k dispozici také X_train a y_train.

V scikit-learn lze MSE modelu vypočítat takto:

MSE_model = mean_squared_error(y_true, y_predicted)

Použijeme funkci mean_squared_error z modulu metrics a předáme jí skutečné hodnoty y_true jako první argument a předpovězené hodnoty modelu y_predicted jako druhý argument.

Pokyny

100 XP
  • Importuj mean_squared_error jako MSE z sklearn.metrics.
  • Natrénuj dt na trénovací množině.
  • Předpověz hodnoty dt na trénovací množině a výsledek ulož do y_pred_train.
  • Vypočítej RMSE modelu dt na trénovací množině a ulož ho do RMSE_train.