From 5fc058a1fc5d83ad53feec936475484aef3800b3 Mon Sep 17 00:00:00 2001
From: Xiangrui Meng <meng@databricks.com>
Date: Wed, 12 Aug 2015 21:33:38 -0700
Subject: [PATCH] [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.
---
.../org/apache/spark/ml/feature/MinMaxScaler.scala | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
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
--
GitLab