aboutsummaryrefslogtreecommitdiff
path: root/mllib/src
diff options
context:
space:
mode:
authorXiangrui Meng <meng@databricks.com>2015-08-12 21:33:38 -0700
committerXiangrui Meng <meng@databricks.com>2015-08-12 21:33:38 -0700
commit5fc058a1fc5d83ad53feec936475484aef3800b3 (patch)
tree2c345b616bff9faf74cf7d91e096b316353dda51 /mllib/src
parenta8ab2634c1eee143a4deaf309204df8add727f9e (diff)
downloadspark-5fc058a1fc5d83ad53feec936475484aef3800b3.tar.gz
spark-5fc058a1fc5d83ad53feec936475484aef3800b3.tar.bz2
spark-5fc058a1fc5d83ad53feec936475484aef3800b3.zip
[SPARK-9917] [ML] add getMin/getMax and doc for originalMin/origianlMax in MinMaxScaler
hhbyyh Author: Xiangrui Meng <meng@databricks.com> Closes #8145 from mengxr/SPARK-9917.
Diffstat (limited to 'mllib/src')
-rw-r--r--mllib/src/main/scala/org/apache/spark/ml/feature/MinMaxScaler.scala10
1 files changed, 9 insertions, 1 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/MinMaxScaler.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/MinMaxScaler.scala
index b30adf3df4..9a473dd237 100644
--- a/mllib/src/main/scala/org/apache/spark/ml/feature/MinMaxScaler.scala
+++ b/mllib/src/main/scala/org/apache/spark/ml/feature/MinMaxScaler.scala
@@ -41,6 +41,9 @@ private[feature] trait MinMaxScalerParams extends Params with HasInputCol with H
val min: DoubleParam = new DoubleParam(this, "min",
"lower bound of the output feature range")
+ /** @group getParam */
+ def getMin: Double = $(min)
+
/**
* upper bound after transformation, shared by all features
* Default: 1.0
@@ -49,6 +52,9 @@ private[feature] trait MinMaxScalerParams extends Params with HasInputCol with H
val max: DoubleParam = new DoubleParam(this, "max",
"upper bound of the output feature range")
+ /** @group getParam */
+ def getMax: Double = $(max)
+
/** Validates and transforms the input schema. */
protected def validateAndTransformSchema(schema: StructType): StructType = {
val inputType = schema($(inputCol)).dataType
@@ -115,6 +121,9 @@ class MinMaxScaler(override val uid: String)
* :: Experimental ::
* Model fitted by [[MinMaxScaler]].
*
+ * @param originalMin min value for each original column during fitting
+ * @param originalMax max value for each original column during fitting
+ *
* TODO: The transformer does not yet set the metadata in the output column (SPARK-8529).
*/
@Experimental
@@ -136,7 +145,6 @@ class MinMaxScalerModel private[ml] (
/** @group setParam */
def setMax(value: Double): this.type = set(max, value)
-
override def transform(dataset: DataFrame): DataFrame = {
val originalRange = (originalMax.toBreeze - originalMin.toBreeze).toArray
val minArray = originalMin.toArray