1. Nauka
    2. /
  2. Kursy
    3. /
  3. Uczenie maszynowe z modelami drzewiastymi w Pythonie
Connected

ćwiczenie

Regresja liniowa a drzewo regresji

W tym ćwiczeniu porównasz wartość RMSE na zbiorze testowym dla modelu dt z wartością uzyskaną przez model regresji liniowej. Model regresji liniowej lr został już zainicjalizowany i wytrenowany na tym samym zbiorze danych co dt.

W twoim obszarze roboczym dostępne są: macierz cech X_test, tablica etykiet y_test, wytrenowany model regresji liniowej lr, funkcja mean_squared_error zaimportowana pod aliasem MSE oraz wartość rmse_dt z poprzedniego ćwiczenia.

Instrukcje

100 XP

  • Wyznacz etykiety dla zbioru testowego za pomocą modelu regresji liniowej (lr) i przypisz wynik do y_pred_lr.

  • Oblicz MSE na zbiorze testowym i przypisz wynik do mse_lr.

  • Oblicz RMSE na zbiorze testowym i przypisz wynik do rmse_lr.