Github user viirya commented on a diff in the pull request:
https://github.com/apache/spark/pull/20146#discussion_r163528221
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/feature/StringIndexerSuite.scala ---
@@ -331,4 +357,51 @@ class StringIndexerSuite
val dfWithIndex = model.transform(dfNoBristol)
assert(dfWithIndex.filter($"CITYIndexed" === 1.0).count == 1)
}
+
+ test("StringIndexer multiple input columns") {
--- End diff --
Good point. I added one test for this.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
