1. Learn
    2. /
  2. Cursuri
    3. /
  3. Обобщённые линейные модели в Python
Connected

Exercițiu

Моделирование с категориальной переменной

В предыдущих упражнениях вы строили модель логистической регрессии с переменной color в качестве объясняющей переменной вместе с width, при этом color рассматривалась как количественная переменная. В этом упражнении вы будете рассматривать color как категориальную переменную: при построении матрицы модели она будет закодирована в 3 переменные с кодировкой 0/1.

Напомним, что кодировка по умолчанию в dmatrix() использует первую группу в качестве референтной. Чтобы отобразить матрицу модели в виде датафрейма, в dmatrix() нужно передать дополнительный аргумент return_type со значением 'dataframe'.

Переменная color имеет следующий естественный порядок:
1: medium light
2: medium
3: medium dark
4: dark

Набор данных crab уже загружен в рабочую среду.

Instrucțiuni 1/4

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