fix style

liancheng · Jan 22, 2015 · 56de571 · 56de571
1 parent d0c5bb8
commit 56de571
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/examples/src/main/python/ml/simple_text_classification_pipeline.py b/examples/src/main/python/ml/simple_text_classification_pipeline.py
@@ -29,28 +29,28 @@
         sc.parallelize([(0L, "a b c d e spark", 1.0),
                         (1L, "b d", 0.0),
                         (2L, "spark f g h", 1.0),
-                        (3L, "hadoop mapreduce", 0.0)]) \
+                        (3L, "hadoop mapreduce", 0.0)])
           .map(lambda x: Row(id=x[0], text=x[1], label=x[2])))
 
     tokenizer = Tokenizer() \
-      .setInputCol("text") \
-      .setOutputCol("words")
+        .setInputCol("text") \
+        .setOutputCol("words")
     hashingTF = HashingTF() \
-      .setInputCol(tokenizer.getOutputCol()) \
-      .setOutputCol("features")
+        .setInputCol(tokenizer.getOutputCol()) \
+        .setOutputCol("features")
     lr = LogisticRegression() \
-      .setMaxIter(10) \
-      .setRegParam(0.01)
+        .setMaxIter(10) \
+        .setRegParam(0.01)
     pipeline = Pipeline() \
-      .setStages([tokenizer, hashingTF, lr])
+        .setStages([tokenizer, hashingTF, lr])
 
     model = pipeline.fit(training)
 
     test = sqlCtx.inferSchema(
         sc.parallelize([(4L, "spark i j k"),
                         (5L, "l m n"),
                         (6L, "mapreduce spark"),
-                        (7L, "apache hadoop")]) \
+                        (7L, "apache hadoop")])
           .map(lambda x: Row(id=x[0], text=x[1])))
 
     for row in model.transform(test).collect():

diff --git a/python/pyspark/ml/__init__.py b/python/pyspark/ml/__init__.py
@@ -18,7 +18,7 @@
 from abc import ABCMeta, abstractmethod, abstractproperty
 
 from pyspark import SparkContext
-from pyspark.sql import SchemaRDD, inherit_doc # TODO: move inherit_doc to Spark Core
+from pyspark.sql import SchemaRDD, inherit_doc  # TODO: move inherit_doc to Spark Core
 from pyspark.ml.param import Param, Params
 from pyspark.ml.util import Identifiable