Spark中的向量Vectors

Spark中的向量Vectors

最近在尝试编写k-means算法的代码,首先需要将TXT数据文件转换为向量,在此过程中,我用到的是Vectors.dense()函数,就将其用法总结如下。
import org.apache.spark.mllib.linalg.Vectors
//将数据集data进行转化,首先根据空格将数据分开然后再将其转化为double类型
val data1=data.map(s=>Vectors.dense(s.split(’ ').map(_.toDouble))
//打印转化后的数据,可见数据已成功转化为向量
data1.foreach(println)

基本用法
主要还是声明或定义向量时:
//定义一个向量,输出结果:[1.0,2.0,3.0]
val v1=Vectors.dense(1.0,2.0,3.0)

猜你喜欢

转载自blog.csdn.net/weixin_42292787/article/details/82974395