Zacznij terazZacznij za darmo

Stacking modeli I

Czas na stacking. Aby wdrożyć to podejście, wykonaj 6 kroków omówionych w poprzednim filmie:

  1. Podziel dane treningowe na dwie części
  2. Wytrenuj kilka modeli na Części 1
  3. Wykonaj predykcje na Części 2
  4. Wykonaj predykcje na danych testowych
  5. Wytrenuj nowy model na Części 2, używając predykcji jako cech
  6. Wykonaj predykcje na danych testowych przy użyciu modelu 2. poziomu

DataFrame'y train i test są już dostępne w twoim środowisku. features to lista kolumn używanych do trenowania na danych z Części 1 – również dostępna w środowisku. Nazwa zmiennej docelowej to „fare_amount".

To ćwiczenie jest częścią kursu

Zwycięstwo w konkursie Kaggle w Pythonie

Zobacz kurs

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

from sklearn.model_selection import train_test_split
from sklearn.ensemble import GradientBoostingRegressor, RandomForestRegressor

# Split train data into two parts
part_1, part_2 = ____(train, test_size=____, random_state=123)

# Train a Gradient Boosting model on Part 1
gb = GradientBoostingRegressor().____(____[features], ____.fare_amount)

# Train a Random Forest model on Part 1
rf = RandomForestRegressor().____(____[features], ____.fare_amount)
Edytuj i uruchom kod