diff --git a/mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala b/mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala index e61997c5934de..474d4731ec0de 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala @@ -130,7 +130,7 @@ class ALSModel private[ml] ( .join(items, dataset(map(itemCol)) === items("id"), "left") .select(outputColumns: _*) // TODO: Just use a dataset("*") - //.select(dataset("*"), prediction) + // .select(dataset("*"), prediction) } override private[ml] def transformSchema(schema: StructType, paramMap: ParamMap): StructType = {