Github user MLnick commented on a diff in the pull request:
https://github.com/apache/spark/pull/17673#discussion_r142218654
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/feature/Word2VecSuite.scala ---
@@ -245,5 +508,28 @@ class Word2VecSuite extends SparkFunSuite with
MLlibTestSparkContext with Defaul
model.transform(ngramDF).collect()
}
+ test("Word2Vec works with input that is non-nullable (NGram) - CBOW") {
--- End diff --
Can this test not just be folded into the one above?
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
