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

cvičení

Měření AUC

Teď, když jsi použil/a křížovou validaci k výpočtu průměrné přesnosti na testovacích datech (po převodu z chyby), je velmi snadné vypočítat jakoukoli jinou metriku, která tě zajímá. Stačí ji předat (nebo seznam metrik) jako argument parametru metrics funkce xgb.cv().

Tvým úkolem v tomto cvičení je vypočítat další běžnou metriku používanou při binární klasifikaci – plochu pod křivkou ("auc"). Stejně jako dříve máš v pracovním prostoru k dispozici churn_data, DMatrix churn_dmatrix a slovník parametrů params.

Pokyny

100 XP
  • Proveď 3-foldovou křížovou validaci s 5 boosting rounds a metrikou "auc".
  • Vypiš sloupec "test-auc-mean" z cv_results.