Skip to content

Commit

Permalink
SPARK-8484. Styling.
Browse files Browse the repository at this point in the history
  • Loading branch information
zapletal-martin committed Jul 10, 2015
1 parent d699506 commit 00c4f5a
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package org.apache.spark.ml.tuning

import org.apache.spark.SparkFunSuite
Expand All @@ -33,7 +33,7 @@ class TrainValidationSplitSuite extends SparkFunSuite with MLlibTestSparkContext
test("train validation with logistic regression") {
val dataset = sqlContext.createDataFrame(
sc.parallelize(generateLogisticInput(1.0, 1.0, 100, 42), 2))

val lr = new LogisticRegression
val lrParamMaps = new ParamGridBuilder()
.addGrid(lr.regParam, Array(0.001, 1000.0))
Expand All @@ -52,7 +52,7 @@ class TrainValidationSplitSuite extends SparkFunSuite with MLlibTestSparkContext
assert(parent.getMaxIter === 10)
assert(cvModel.avgMetrics.length === lrParamMaps.length)
}

test("train validation with linear regression") {
val dataset = sqlContext.createDataFrame(
sc.parallelize(LinearDataGenerator.generateLinearInput(
Expand Down Expand Up @@ -82,7 +82,7 @@ class TrainValidationSplitSuite extends SparkFunSuite with MLlibTestSparkContext
assert(parent2.getMaxIter === 10)
assert(cvModel2.avgMetrics.length === lrParamMaps.length)
}

test("validateParams should check estimatorParamMaps") {
import TrainValidationSplitSuite._

Expand Down

0 comments on commit 00c4f5a

Please sign in to comment.