1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Extreme Gradient Boosting with XGBoost
Connected

cvičení

Ladění max_depth

V tomto cvičení budeš ladit parametr max_depth, který určuje maximální hloubku, do které může každý strom v jednom boosting kole vyrůst. Menší hodnoty vedou k mělčím stromům, větší hodnoty pak ke hlubším.

Pokyny

100 XP
  • Vytvoř seznam max_depths, který bude obsahovat následující hodnoty parametru "max_depth": 2, 5, 10 a 20.
  • Pomocí cyklu for projdi seznam max_depths.
  • V každé iteraci cyklu for systematicky měň hodnotu "max_depth" a proveď 2-násobnou křížovou validaci s předčasným zastavením (5 kol), 10 boosting koly, metrikou "rmse" a hodnotou seed nastavenou na 123. Výstup zajisti ve formátu DataFrame.