Fix comments

apache · Jan 17, 2018 · b4f2c71 · b4f2c71
1 parent 85d0db0
commit b4f2c71
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/examples/src/main/python/ml/vector_size_hint_example.py b/examples/src/main/python/ml/vector_size_hint_example.py
@@ -41,12 +41,14 @@
         size=3)
 
     datasetWithSize = sizeHint.transform(dataset)
+    print("Rows where 'userFeatures' is not the right size are filtered out")
     datasetWithSize.show(truncate=False)
 
     assembler = VectorAssembler(
         inputCols=["hour", "mobile", "userFeatures"],
         outputCol="features")
 
+    # This dataframe can be used by used by downstream transformers as before
     output = assembler.transform(datasetWithSize)
     print("Assembled columns 'hour', 'mobile', 'userFeatures' to vector column 'features'")
     output.select("features", "clicked").show(truncate=False)

diff --git a/examples/src/main/scala/org/apache/spark/examples/ml/VectorSizeHintExample.scala b/examples/src/main/scala/org/apache/spark/examples/ml/VectorSizeHintExample.scala
@@ -44,13 +44,16 @@ object VectorSizeHintExample {
       .setSize(3)
 
     val datasetWithSize = sizeHint.transform(dataset)
+    println("Rows where 'userFeatures' is not the right size are filtered out")
     datasetWithSize.show(false)
 
     val assembler = new VectorAssembler()
       .setInputCols(Array("hour", "mobile", "userFeatures"))
       .setOutputCol("features")
 
+    // This dataframe can be used by used by downstream transformers as before
     val output = assembler.transform(datasetWithSize)
+    println("Assembled columns 'hour', 'mobile', 'userFeatures' to vector column 'features'")
     output.select("features", "clicked").show(false)
     // $example off$