开始使用免费开始使用

训练逻辑回归模型

逻辑回归是一种简单但非常强大的分类模型,适用于多种场景。现在,您将对电信流失数据集的训练集拟合一个逻辑回归模型,然后在未见过的测试集上预测标签。之后,您将计算模型预测的准确率。

accuracy_score 函数已导入,且来自 sklearnLogisticRegression 实例已初始化为 logreg。您之前构建的训练集和测试集已分别加载为特征 train_Xtest_X,以及目标变量 train_Ytest_Y

本练习是课程的一部分

Python 营销中的机器学习

查看课程

练习说明

  • 在训练数据上拟合逻辑回归模型。
  • 为测试数据预测是否流失的标签。
  • 计算测试数据上的准确率。
  • 将测试准确率四舍五入到 4 位小数并打印。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Fit logistic regression on training data
logreg.___(train_X, train_Y)

# Predict churn labels on testing data
pred_test_Y = ___.predict(test_X)

# Calculate accuracy score on testing data
test_accuracy = ___(test_Y, pred_test_Y)

# Print test accuracy score rounded to 4 decimals
print('Test accuracy:', ___(test_accuracy, 4))
编辑并运行代码