Github user shahidki31 commented on a diff in the pull request:
https://github.com/apache/spark/pull/21509#discussion_r194144115
--- Diff:
mllib/src/main/scala/org/apache/spark/ml/clustering/PowerIterationClustering.scala
---
@@ -166,6 +166,7 @@ class PowerIterationClustering private[clustering] (
val w = if (!isDefined(weightCol) || $(weightCol).isEmpty) {
lit(1.0)
} else {
+ SchemaUtils.checkColumnTypes(dataset.schema, $(weightCol),
Seq(FloatType, DoubleType))
--- End diff --
Yes. To be consistent with the test case "supported input types" and the
previous PR, I was checking only for 'FloatType' and 'DoubleType' for
similarity column. I have modified the code.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
