Skip to content

Commit 54d2d08

Browse files
committed
update examples to use regression data instead of classification data
1 parent cad4f71 commit 54d2d08

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

examples/src/main/java/org/apache/spark/examples/ml/JavaTrainValidationSplitExample.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public static void main(String[] args) {
4444
JavaSparkContext jsc = new JavaSparkContext(conf);
4545
SQLContext jsql = new SQLContext(jsc);
4646

47-
DataFrame data = jsql.read().format("libsvm").load("data/mllib/sample_libsvm_data.txt");
47+
DataFrame data = jsql.read().format("libsvm").load("data/mllib/sample_linear_regression_data.txt");
4848

4949
// Prepare training and test data.
5050
DataFrame[] splits = data.randomSplit(new double [] {0.9, 0.1}, 12345);

examples/src/main/scala/org/apache/spark/examples/ml/TrainValidationSplitExample.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ object TrainValidationSplitExample {
4040
val sqlContext = new SQLContext(sc)
4141

4242
// Prepare training and test data.
43-
val data = sqlContext.read.format("libsvm").load("data/mllib/sample_libsvm_data.txt")
43+
val data = sqlContext.read.format("libsvm").load("data/mllib/sample_linear_regression_data.txt")
4444
val Array(training, test) = data.randomSplit(Array(0.9, 0.1), seed = 12345)
4545

4646
val lr = new LinearRegression()

0 commit comments

Comments
 (0)