5、电信客户流失分析与预测

电信客户流失分析与预测

1. 特征提取与转换

在完成使用 StringIndexer 将分类标签转换为数值后，下一步是提取特征。代码如下：

val featureCols = Array("account_length", "iplanIndex",
                        "num_voice_mail", "total_day_mins",
                        "total_day_calls", "total_evening_mins",
                        "total_evening_calls", "total_night_mins",
                        "total_night_calls", "total_international_mins",
                        "total_international_calls",
                        "total_international_num_calls")

接着，使用 VectorAssembler 将特征转换为特征向量，代码如下：

val assembler = new VectorAssembler()
    .setInputCols(featureCols)
    .setOutputCol("features")