spark gbdt 自定义阈值 取出模型几率，并转换label

在你们使用spark 的时候，会发现，ML库下的模型生成的几率是一个Vector，那么如何将这个Vector的几率为1的那一列取出呢？而且自定义阈值，按照这个阈值切分获得label的0，1类呢？ 这时候udf函数就派上了很好的用场，废话很少说，直接上代码web 取出Vector 的第n列，生成新的dataframe 切分Vector获得每一列的值，形如调用下面的代码生成features,对feat

>>阅读原文<<