如何将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

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

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