[SPARK-6308] [MLlib] [Sql] Override TypeName in VectorUDT and MatrixUDT
Author: MechCoder <manojkumarsivaraj334@gmail.com> Closes #5118 from MechCoder/spark-6308 and squashes the following commits: 6c8ffab [MechCoder] Add test for simpleString b966242 [MechCoder] [SPARK-6308] [MLlib][Sql] VectorUDT is displayed as vecto in dtypes
This commit is contained in:
parent
9f3273bd9c
commit
474d1320c9
|
@ -187,6 +187,8 @@ private[spark] class MatrixUDT extends UserDefinedType[Matrix] {
|
|||
|
||||
override def hashCode(): Int = 1994
|
||||
|
||||
override def typeName: String = "matrix"
|
||||
|
||||
private[spark] override def asNullable: MatrixUDT = this
|
||||
}
|
||||
|
||||
|
|
|
@ -185,6 +185,8 @@ private[spark] class VectorUDT extends UserDefinedType[Vector] {
|
|||
|
||||
override def hashCode: Int = 7919
|
||||
|
||||
override def typeName: String = "vector"
|
||||
|
||||
private[spark] override def asNullable: VectorUDT = this
|
||||
}
|
||||
|
||||
|
|
|
@ -436,5 +436,7 @@ class MatricesSuite extends FunSuite {
|
|||
Seq(dm1, dm2, dm3, sm1, sm2, sm3).foreach {
|
||||
mat => assert(mat.toArray === mUDT.deserialize(mUDT.serialize(mat)).toArray)
|
||||
}
|
||||
assert(mUDT.typeName == "matrix")
|
||||
assert(mUDT.simpleString == "matrix")
|
||||
}
|
||||
}
|
||||
|
|
|
@ -187,6 +187,8 @@ class VectorsSuite extends FunSuite {
|
|||
for (v <- Seq(dv0, dv1, sv0, sv1)) {
|
||||
assert(v === udt.deserialize(udt.serialize(v)))
|
||||
}
|
||||
assert(udt.typeName == "vector")
|
||||
assert(udt.simpleString == "vector")
|
||||
}
|
||||
|
||||
test("fromBreeze") {
|
||||
|
|
Loading…
Reference in a new issue