Skip to content

Commit

Permalink
Remove the inhefited docs
Browse files Browse the repository at this point in the history
  • Loading branch information
yu-iskw committed Jul 1, 2015
1 parent e9532e1 commit d3a79f7
Showing 1 changed file with 0 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -114,21 +114,13 @@ class KMeansModel private[ml] (
copyValues(copied, extra)
}

/**
* Transforms the input dataset.
*/
override def transform(dataset: DataFrame): DataFrame = {
dataset.select(
dataset("*"),
callUDF(predict _, IntegerType, col($(featuresCol))).as($(predictionCol))
)
}

/**
* :: DeveloperApi ::
*
* Derives the output schema from the input schema.
*/
override def transformSchema(schema: StructType): StructType = {
validateAndTransformSchema(schema)
}
Expand Down

0 comments on commit d3a79f7

Please sign in to comment.