1. 学ぶ
  2. /
  3. コース
  4. /
  5. Узагальнені лінійні моделі в Python

Connected

演習

Моделювання з категоріальною змінною

У попередніх вправах ви підігнали модель логістичної регресії з color як пояснювальною змінною разом із width, де ви розглядали color як кількісну змінну. У цій вправі ви розглядатимете color як категоріальну змінну, що під час побудови матриці моделі закодує color у 3 змінні з кодуванням 0/1.

Згадайте, що типове кодування в dmatrix() використовує першу групу як еталонну. Щоб переглянути матрицю моделі як датафрейм, додатковий аргумент у dmatrix(), а саме return_type, потрібно встановити в 'dataframe'.

Змінна color має природний порядок такий:
1: medium light
2: medium
3: medium dark
4: dark

Набір даних crab попередньо завантажено в робочому середовищі.

指示1 / 4

undefined XP
    1
    2
    3
    4
  • Побудуйте model_matrix зі змінною color. color слід вважати категоріальною, а еталонну групу встановити як 4 за допомогою функції Treatment().
  • Підіженіть і виведіть результати логістичної моделі з y як відгуком (response).