How to convert Spark Dense Vector to String and back to Vector in Scala?

Question

I have Dense Vector, I would like to convert vector into string (to save CSV) and convert string back to Dense Vector when load.

More detail

val dense_vec = Vectors.dense(1.0, 2.0, 3.0)
dense_vec: org.apache.spark.mllib.linalg.Vector = [1.0,2.0,3.0]

val str_dense_vec = dense_vec.toString
str_dense_vec: String = [1.0,2.0,3.0]

I want to convert str_dense_vec as type String into org.apache.spark.mllib.linalg.Vector

Thang Nguyen · Accepted Answer · 2017-11-08 09:58:19Z

1

You can create Double Array from String, then use dense method of org.apache.spark.mllib.linalg.Vector.

Vectors.dense(str_dense_vec.drop(1).dropRight(1).split(',').map(_.toDouble))

link api.

answered Nov 8, 2017 at 9:58

Thang Nguyen

1,1209 silver badges19 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to convert Spark Dense Vector to String and back to Vector in Scala?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related