Skip to content

Commit

Permalink
disable cv doctest
Browse files Browse the repository at this point in the history
  • Loading branch information
mengxr committed May 7, 2015
1 parent 14502d5 commit 5db7e5b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/pyspark/ml/tuning.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,10 +103,10 @@ class CrossValidator(Estimator):
>>> grid = ParamGridBuilder().addGrid(lr.maxIter, [0, 1, 5]).build()
>>> evaluator = BinaryClassificationEvaluator()
>>> cv = CrossValidator(estimator=lr, estimatorParamMaps=grid, evaluator=evaluator)
>>> cvModel = cv.fit(dataset)
>>> expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
>>> cvModel.transform(dataset).collect() == expected.collect()
True
>>> # SPARK-7432: The following test is flaky.
>>> # cvModel = cv.fit(dataset)
>>> # expected = lr.fit(dataset, {lr.maxIter: 5}).transform(dataset)
>>> # cvModel.transform(dataset).collect() == expected.collect()
"""

# a placeholder to make it appear in the generated doc
Expand Down

0 comments on commit 5db7e5b

Please sign in to comment.