Skip to content

Commit ad678e9

Browse files
committed
more doc for Transformer
1 parent 4306ed4 commit ad678e9

File tree

3 files changed

+18
-4
lines changed

3 files changed

+18
-4
lines changed

mllib/src/main/scala/org/apache/spark/ml/Model.scala

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,9 @@ package org.apache.spark.ml
2020
import org.apache.spark.ml.param.ParamMap
2121

2222
/**
23-
* A fitted model.
23+
* A fitted model, i.e., a [[Transformer]] produced by an [[Estimator]].
24+
*
25+
* @tparam M model type
2426
*/
2527
abstract class Model[M <: Model[M]] extends Transformer {
2628
/**

mllib/src/main/scala/org/apache/spark/ml/Pipeline.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ abstract class PipelineStage extends Serializable with Logging {
3434
def transform(schema: StructType, paramMap: ParamMap): StructType
3535

3636
/**
37-
* Drives the output schema from the input schema and parameters, optionally with logging.
37+
* Derives the output schema from the input schema and parameters, optionally with logging.
3838
*/
3939
protected def transform(schema: StructType, paramMap: ParamMap, logging: Boolean): StructType = {
4040
if (logging) {

mllib/src/main/scala/org/apache/spark/ml/Transformer.scala

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,20 +48,32 @@ abstract class Transformer extends PipelineStage with Params {
4848
}
4949

5050
/**
51-
* Transforms the dataset with provided parameter map.
51+
* Transforms the dataset with provided parameter map as additional parameters.
5252
* @param dataset input dataset
53-
* @param paramMap parameters
53+
* @param paramMap additional parameters, overwrite embedded params
5454
* @return transformed dataset
5555
*/
5656
def transform(dataset: SchemaRDD, paramMap: ParamMap): SchemaRDD
5757

5858
// Java-friendly versions of transform.
5959

60+
/**
61+
* Transforms the dataset with optional parameters.
62+
* @param dataset input datset
63+
* @param paramPairs optional list of param pairs, overwrite embedded params
64+
* @return transformed dataset
65+
*/
6066
@varargs
6167
def transform(dataset: JavaSchemaRDD, paramPairs: ParamPair[_]*): JavaSchemaRDD = {
6268
transform(dataset.schemaRDD, paramPairs: _*).toJavaSchemaRDD
6369
}
6470

71+
/**
72+
* Transforms the dataset with provided parameter map as additional parameters.
73+
* @param dataset input dataset
74+
* @param paramMap additional parameters, overwrite embedded params
75+
* @return transformed dataset
76+
*/
6577
def transform(dataset: JavaSchemaRDD, paramMap: ParamMap): JavaSchemaRDD = {
6678
transform(dataset.schemaRDD, paramMap).toJavaSchemaRDD
6779
}

0 commit comments

Comments
 (0)