1. Learn
    2. /
  2. Cursuri
    3. /
  3. Machine Learning cu modele bazate pe arbori în Python
Connected

exercițiu

Regresie liniară vs. arbore de regresie

În acest exercițiu, vei compara RMSE-ul pe setul de testare al lui dt cu cel obținut de un model de regresie liniară. Am instanțiat deja un model de regresie liniară lr și l-am antrenat pe același set de date ca și dt.

Matricea de caracteristici X_test, vectorul de etichete y_test, modelul de regresie liniară antrenat lr, funcția mean_squared_error importată sub aliasul MSE și rmse_dt din exercițiul anterior sunt disponibile în spațiul tău de lucru.

Instrucțiuni

100 XP

  • Prezice etichetele setului de testare folosind modelul de regresie liniară (lr) și atribuie rezultatul variabilei y_pred_lr.

  • Calculează MSE-ul pe setul de testare și atribuie rezultatul variabilei mse_lr.

  • Calculează RMSE-ul pe setul de testare și atribuie rezultatul variabilei rmse_lr.