1. Nauka
    2. /
  2. Kursy
    3. /
  3. Extreme Gradient Boosting with XGBoost
Connected

ćwiczenie

Dostrajanie max_depth

W tym ćwiczeniu dostroisz parametr max_depth, który określa maksymalną głębokość, jaką może osiągnąć każde drzewo w danej rundzie boostingu. Mniejsze wartości dają płytsze drzewa, a większe – głębsze.

Instrukcje

100 XP
  • Utwórz listę o nazwie max_depths, która będzie przechowywać następujące wartości "max_depth": 2, 5, 10 i 20.
  • Iteruj po liście max_depths za pomocą pętli for.
  • W każdej iteracji pętli for systematycznie zmieniaj wartość "max_depth" i wykonuj 2-krotną walidację krzyżową z wczesnym zatrzymaniem (5 rund), 10 rundami boostingu, metryką "rmse" i ziarnem losowości (seed) równym 123. Zadbaj o to, aby wynik był zwracany jako DataFrame.