aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/ml/tuning.py
diff options
context:
space:
mode:
authorXiangrui Meng <meng@databricks.com>2015-05-06 22:29:07 -0700
committerXiangrui Meng <meng@databricks.com>2015-05-06 22:29:07 -0700
commit773aa25252f29ca25dbb1ee495b530557fe79405 (patch)
tree84e7c72da7a84ae89a6e1155b25cf23c23a92e04 /python/pyspark/ml/tuning.py
parent14502d5e569f8460d3e2c973a4277441307433cc (diff)
downloadspark-773aa25252f29ca25dbb1ee495b530557fe79405.tar.gz
spark-773aa25252f29ca25dbb1ee495b530557fe79405.tar.bz2
spark-773aa25252f29ca25dbb1ee495b530557fe79405.zip
[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
Diffstat (limited to 'python/pyspark/ml/tuning.py')
-rw-r--r--python/pyspark/ml/tuning.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/python/pyspark/ml/tuning.py b/python/pyspark/ml/tuning.py
index f6cf2c3439..1e04c37fca 100644
--- a/python/pyspark/ml/tuning.py
+++ b/python/pyspark/ml/tuning.py
@@ -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