Fix linter issues

jeanlyn · Apr 21, 2015 · 9a8ce22 · 9a8ce22
1 parent 8b8a6d2
commit 9a8ce22
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/python/pyspark/ml/tuning.py b/python/pyspark/ml/tuning.py
@@ -17,6 +17,7 @@
 
 __all__ = ['ParamGridBuilder']
 
+
 class ParamGridBuilder(object):
     """
     Builder for a param grid used in grid search-based model selection.
@@ -66,18 +67,17 @@ def build(self):
     lr = LogisticRegression()
     grid_test.addGrid(lr.regParam, [1.0, 2.0, 3.0])
     grid_test.addGrid(lr.maxIter, [1, 5])
-    grid_test.addGrid(lr.featuresCol, ['f'])
+    grid_test.addGrid(lr.inputCol, ['f'])
     grid_test.baseOn({lr.labelCol: 'l'})
-    grid_test.baseOn([lr.predictionCol, 'p'])
+    grid_test.baseOn([lr.outputCol, 'p'])
     grid = grid_test.build()
     expected = [
-            {lr.regParam: 1.0, lr.featuresCol: 'f', lr.maxIter: 1, lr.labelCol: 'l', lr.predictionCol: 'p'},
-            {lr.regParam: 2.0, lr.featuresCol: 'f', lr.maxIter: 1, lr.labelCol: 'l', lr.predictionCol: 'p'},
-            {lr.regParam: 3.0, lr.featuresCol: 'f', lr.maxIter: 1, lr.labelCol: 'l', lr.predictionCol: 'p'},
-            {lr.regParam: 1.0, lr.featuresCol: 'f', lr.maxIter: 5, lr.labelCol: 'l', lr.predictionCol: 'p'},
-            {lr.regParam: 2.0, lr.featuresCol: 'f', lr.maxIter: 5, lr.labelCol: 'l', lr.predictionCol: 'p'},
-            {lr.regParam: 3.0, lr.featuresCol: 'f', lr.maxIter: 5, lr.labelCol: 'l', lr.predictionCol: 'p'}
-        ]
+        {lr.regParam: 1.0, lr.inputCol: 'f', lr.maxIter: 1, lr.labelCol: 'l', lr.outputCol: 'p'},
+        {lr.regParam: 2.0, lr.inputCol: 'f', lr.maxIter: 1, lr.labelCol: 'l', lr.outputCol: 'p'},
+        {lr.regParam: 3.0, lr.inputCol: 'f', lr.maxIter: 1, lr.labelCol: 'l', lr.outputCol: 'p'},
+        {lr.regParam: 1.0, lr.inputCol: 'f', lr.maxIter: 5, lr.labelCol: 'l', lr.outputCol: 'p'},
+        {lr.regParam: 2.0, lr.inputCol: 'f', lr.maxIter: 5, lr.labelCol: 'l', lr.outputCol: 'p'},
+        {lr.regParam: 3.0, lr.inputCol: 'f', lr.maxIter: 5, lr.labelCol: 'l', lr.outputCol: 'p'}]
 
     for a, b in zip(grid, expected):
         if a != b: