Github user mengxr commented on a diff in the pull request:
https://github.com/apache/spark/pull/21195#discussion_r185971385
--- Diff: mllib/src/main/scala/org/apache/spark/ml/util/SchemaUtils.scala
---
@@ -101,4 +102,17 @@ private[spark] object SchemaUtils {
require(!schema.fieldNames.contains(col.name), s"Column ${col.name}
already exists.")
StructType(schema.fields :+ col)
}
+
+ /**
+ * Check whether the given column in the schema is one of the supporting
vector type: Vector,
+ * Array[Dloat]. Array[Double]
--- End diff --
nit: Float
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]