1. Nauka
    2. /
  2. Kursy
    3. /
  3. Wprowadzenie do analizy predykcyjnej w Pythonie
Connected

ćwiczenie

Ocena modelu na zbiorze treningowym i testowym

Funkcja auc_train_test oblicza AUC modelu zbudowanego na zbiorze train i ocenianego na zbiorze test: 

auc_train, auc_test = auc_train_test(variables, target, train, test)

gdzie variables to lista nazw zmiennych użytych w modelu.

W tym ćwiczeniu zastosujesz tę funkcję i sprawdzisz, czy wartości AUC dla zbioru treningowego i testowego są do siebie zbliżone.

Instrukcje

100 XP
  • Zbiór basetable jest już wczytany. Podziel go tak, aby zbiór treningowy zawierał 70% danych, i zadbaj o to, by zbiór treningowy i testowy miały jednakową częstość występowania klasy docelowej.
  • Oblicz AUC na zbiorze treningowym i testowym dla modelu używającego "age" i "gender_F" jako predyktorów – skorzystaj z funkcji auc_train_test.