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

ćwiczenie

Ocena optymalnego lasu

W tym ostatnim ćwiczeniu kursu ocenisz RMSE zbioru testowego dla optymalnego modelu z grid_rf.

Zbiór danych jest już wczytany i przetworzony – podzielony na 80% danych treningowych i 20% testowych. W środowisku dostępne są X_test, y_test oraz funkcja mean_squared_error z sklearn.metrics pod aliasem MSE. Wczytany jest również wytrenowany obiekt GridSearchCV o nazwie grid_rf, który był tworzony w poprzednim ćwiczeniu. Zwróć uwagę, że grid_rf był trenowany w następujący sposób:

grid_rf.fit(X_train, y_train)

Instrukcje

100 XP

  • Zaimportuj mean_squared_error jako MSE z sklearn.metrics.

  • Wyodrębnij najlepszy estymator z grid_rf i przypisz go do zmiennej best_model.

  • Wygeneruj predykcje dla zbioru testowego za pomocą best_model i przypisz wynik do y_pred.

  • Oblicz RMSE best_model na zbiorze testowym.