Skip to content
Snippets Groups Projects
Commit adb54836 authored by JeremyNixon's avatar JeremyNixon Committed by Sean Owen
Browse files

[SPARK-13312][MLLIB] Update java train-validation-split example in ml-guide

Response to JIRA https://issues.apache.org/jira/browse/SPARK-13312.

This contribution is my original work and I license the work to this project.

Author: JeremyNixon <jnixon2@gmail.com>

Closes #11199 from JeremyNixon/update_train_val_split_example.
parent 56d49397
No related branches found
No related tags found
No related merge requests found
......@@ -870,7 +870,7 @@ import org.apache.spark.ml.regression.LinearRegression
import org.apache.spark.ml.tuning.{ParamGridBuilder, TrainValidationSplit}
// Prepare training and test data.
val data = sqlContext.read.format("libsvm").load("data/mllib/sample_libsvm_data.txt")
val data = sqlContext.read.format("libsvm").load("data/mllib/sample_linear_regression_data.txt")
val Array(training, test) = data.randomSplit(Array(0.9, 0.1), seed = 12345)
val lr = new LinearRegression()
......@@ -913,7 +913,7 @@ import org.apache.spark.ml.regression.LinearRegression;
import org.apache.spark.ml.tuning.*;
import org.apache.spark.sql.DataFrame;
DataFrame data = jsql.read().format("libsvm").load("data/mllib/sample_libsvm_data.txt");
DataFrame data = jsql.read().format("libsvm").load("data/mllib/sample_linear_regression_data.txt");
// Prepare training and test data.
DataFrame[] splits = data.randomSplit(new double[] {0.9, 0.1}, 12345);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment