diff options
author | yinxusen <yinxusen@gmail.com> | 2016-06-08 09:18:04 +0100 |
---|---|---|
committer | Sean Owen <sowen@cloudera.com> | 2016-06-08 09:18:04 +0100 |
commit | 87706eb66cd1370862a1f8ea447484c80969e45f (patch) | |
tree | 5084c7fa634744777aadc3886d48de4d5b2e8839 /mllib/src/test | |
parent | 91fbc880b69bddcf5310afecc49df1102408e1f3 (diff) | |
download | spark-87706eb66cd1370862a1f8ea447484c80969e45f.tar.gz spark-87706eb66cd1370862a1f8ea447484c80969e45f.tar.bz2 spark-87706eb66cd1370862a1f8ea447484c80969e45f.zip |
[SPARK-15793][ML] Add maxSentenceLength for ml.Word2Vec
## What changes were proposed in this pull request?
https://issues.apache.org/jira/browse/SPARK-15793
Word2vec in ML package should have maxSentenceLength method for feature parity.
## How was this patch tested?
Tested with Spark unit test.
Author: yinxusen <yinxusen@gmail.com>
Closes #13536 from yinxusen/SPARK-15793.
Diffstat (limited to 'mllib/src/test')
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/ml/feature/Word2VecSuite.scala | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/Word2VecSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/Word2VecSuite.scala index 280a36f56e..16c74f6785 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/feature/Word2VecSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/feature/Word2VecSuite.scala @@ -191,6 +191,7 @@ class Word2VecSuite extends SparkFunSuite with MLlibTestSparkContext with Defaul .setSeed(42L) .setStepSize(0.01) .setVectorSize(100) + .setMaxSentenceLength(500) testDefaultReadWrite(t) } |