预测新客户是否会流失
正如您在视频中所看到的,使用 sklearn 训练模型的步骤如下:
- 导入需要的模型——这里以一个 支持向量分类器 为例:
from sklearn.svm import SVC
- 实例化模型:
svc = SVC()
- 将模型"拟合"到数据上进行训练:
svc.fit(telco['data'], telco['target'])
这里,第一个参数是特征,第二个参数是我们要预测的标签——客户是否会流失。模型拟合完成后,您可以使用模型的 .predict() 方法来预测新客户的标签。
无论使用哪种模型,这个流程都适用,而 sklearn 提供了很多模型可选!在本练习中,您将使用 LogisticRegression。
本练习是课程的一部分