fix df.drop in VectorAssembler (#117)

fix df.drop in VectorAssembler to return correct DataFrame
2017-09-02 04:51:05 +08:00 · 2017-09-02 04:51:05 +08:00 · 19c41464c7
commit 19c41464c7
parent 6ec83fd0f7
1 changed files with 0 additions and 1 deletions
--- a/src/main/scala/com/databricks/spark/sql/perf/mllib/feature/VectorAssembler.scala
+++ b/src/main/scala/com/databricks/spark/sql/perf/mllib/feature/VectorAssembler.scala
@ -34,7 +34,6 @@ object VectorAssembler extends BenchmarkAlgorithm with TestFromTraining {
      df = df.withColumn(colName, sliceVec(col("features"), lit(fromIndex), lit(untilIndex)))
    }
    df.drop(col("features"))
-    df
  }

  override def getPipelineStage(ctx: MLBenchContext): PipelineStage = {