Skip to content
Snippets Groups Projects
Commit 773aa252 authored by Xiangrui Meng's avatar Xiangrui Meng
Browse files

[SPARK-7432] [MLLIB] disable cv doctest

Temporarily disable flaky doctest for CrossValidator. jkbradley

Author: Xiangrui Meng <meng@databricks.com>

Closes #5962 from mengxr/disable-pyspark-cv-test and squashes the following commits:

5db7e5b [Xiangrui Meng] disable cv doctest
parent 14502d5e
No related branches found
No related tags found
No related merge requests found
...@@ -103,10 +103,10 @@ class CrossValidator(Estimator): ...@@ -103,10 +103,10 @@ class CrossValidator(Estimator):
>>> grid = ParamGridBuilder().addGrid(lr.maxIter, [0, 1, 5]).build() >>> grid = ParamGridBuilder().addGrid(lr.maxIter, [0, 1, 5]).build()
>>> evaluator = BinaryClassificationEvaluator() >>> evaluator = BinaryClassificationEvaluator()
>>> cv = CrossValidator(estimator=lr, estimatorParamMaps=grid, evaluator=evaluator) >>> cv = CrossValidator(estimator=lr, estimatorParamMaps=grid, evaluator=evaluator)
>>> cvModel = cv.fit(dataset) >>> # SPARK-7432: The following test is flaky.
>>> expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset) >>> # cvModel = cv.fit(dataset)
>>> cvModel.transform(dataset).collect() == expected.collect() >>> # expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
True >>> # cvModel.transform(dataset).collect() == expected.collect()
""" """
# a placeholder to make it appear in the generated doc # a placeholder to make it appear in the generated doc
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment