Логистическая регрессия с помощью logit()
Для логистической регрессии из statsmodels.formula.api потребуется ещё одна функция — logit(). Она принимает те же аргументы, что и ols(): formula и data. Затем для обучения модели на данных используется метод .fit().
В этом упражнении вы смоделируете, как длительность отношений с клиентом влияет на отток.
Набор данных churn доступен.
Это упражнение является частью курса
Введение в регрессию с statsmodels на Python
Инструкции к упражнению
- Импортируйте функцию
logit()изstatsmodels.formula.api. - Постройте логистическую регрессию
has_churnedв зависимости отtime_since_first_purchase, используя набор данныхchurn. Результат сохраните в переменнуюmdl_churn_vs_relationship. - Выведите на экран параметры обученной модели.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Import logit
____
# Fit a logistic regression of churn vs. length of relationship using the churn dataset
mdl_churn_vs_relationship = ____
# Print the parameters of the fitted model
print(____)