如何将linalg.Vector转换为regression.LabeledPoint格式?

我在spark-shell中尝试实现一个简单的机器学习代码,当我尝试提供一个csv文件时,它要求使用libsvm格式,所以我使用了phraug库来将我的数据集转换为所需格式。虽然这有效,但我还需要对数据进行归一化处理,因此我使用了标准化缩放器来转换数据。这也运行得很好,接下来的步骤是训练机器,为此我使用了SVMWithSGD模型。但当我尝试训练时,我不断收到以下错误

    error: type mismatch;    found: org.apache.spark.rdd.RDD[(Double,org.apache.spark.mllib.linalg.Vector)]    required: org.apache.spark.rdd.RDD[org.apache.spark.mllib.regression.LabeledPoint]

我明白这是一个兼容性问题,可以使用Vector.dense函数,但我不想再次拆分它,我不理解的是,是否有直接的方法可以让我在训练方法中使用它?附注:为了帮助您理解,数据目前看起来像这样

    (0.0,[0.03376345160534202,-0.6339809012492886,-6.719697792783955,-6.719697792783965,-6.30231507117855,-8.72828614492483,0.03884804438718658,0.3041969425433718])    (0.0,[0.2535328275090413,-0.8780294632355746,-6.719697792783955,-6.719697792783965,-6.30231507117855,-8.72828614492483,0.26407233411369857,0.3041969425433718])

回答:

假设你的RDD[Double, Vector]被称为vectorRDD

val labeledPointRDD = vectorRDD map {   case (label, vector) => LabeledPoint(label, vector) }

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注