Skip to content

Commit 216d199

Browse files
committed
fixed after sql datatypes PR got merged
1 parent f549e34 commit 216d199

File tree

8 files changed

+14
-2
lines changed

8 files changed

+14
-2
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,13 @@
1818
package org.apache.spark.examples.ml
1919

2020
import org.apache.spark.{SparkConf, SparkContext}
21-
import org.apache.spark.SparkContext._
2221
import org.apache.spark.ml.classification.{Classifier, ClassifierParams, ClassificationModel}
2322
import org.apache.spark.ml.param.{Params, IntParam, ParamMap}
2423
import org.apache.spark.mllib.linalg.{BLAS, Vector, Vectors}
2524
import org.apache.spark.mllib.regression.LabeledPoint
2625
import org.apache.spark.sql.{SchemaRDD, Row, SQLContext}
2726

27+
2828
/**
2929
* A simple example demonstrating how to write your own learning algorithm using Estimator,
3030
* Transformer, and other abstractions.

mllib/src/main/scala/org/apache/spark/ml/classification/Classifier.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@ import org.apache.spark.ml.param.{Params, ParamMap, HasRawPredictionCol}
2323
import org.apache.spark.mllib.linalg.{Vector, VectorUDT}
2424
import org.apache.spark.sql._
2525
import org.apache.spark.sql.catalyst.analysis.Star
26+
import org.apache.spark.sql.types.{DataType, StructType}
27+
2628

2729
/**
2830
* :: DeveloperApi ::

mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,11 @@ import org.apache.spark.mllib.linalg.{BLAS, Vector, Vectors}
2424
import org.apache.spark.sql._
2525
import org.apache.spark.sql.Dsl._
2626
import org.apache.spark.sql.types.{DoubleType, StructField, StructType}
27+
import org.apache.spark.sql.catalyst.analysis.Star
28+
import org.apache.spark.sql.catalyst.dsl._
2729
import org.apache.spark.storage.StorageLevel
2830

31+
2932
/**
3033
* Params for logistic regression.
3134
*/

mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,8 @@ import org.apache.spark.ml.param.{HasProbabilityCol, ParamMap, Params}
2222
import org.apache.spark.mllib.linalg.{Vector, VectorUDT}
2323
import org.apache.spark.sql._
2424
import org.apache.spark.sql.catalyst.analysis.Star
25+
import org.apache.spark.sql.types.{DataType, StructType}
26+
2527

2628
/**
2729
* Params for probabilistic classification.

mllib/src/main/scala/org/apache/spark/ml/evaluation/BinaryClassificationEvaluator.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ import org.apache.spark.sql.{DataFrame, Row}
2525
import org.apache.spark.mllib.linalg.{Vector, VectorUDT}
2626
import org.apache.spark.sql.types.DoubleType
2727

28+
2829
/**
2930
* :: AlphaComponent ::
3031
*

mllib/src/main/scala/org/apache/spark/ml/impl/estimator/Predictor.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ import org.apache.spark.mllib.regression.LabeledPoint
2525
import org.apache.spark.rdd.RDD
2626
import org.apache.spark.sql._
2727
import org.apache.spark.sql.catalyst.analysis.Star
28+
import org.apache.spark.sql.types.{DataType, DoubleType, StructType}
2829

2930

3031
/**

mllib/src/main/scala/org/apache/spark/ml/param/params.scala

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,8 @@ import java.lang.reflect.Modifier
2424

2525
import org.apache.spark.annotation.{AlphaComponent, DeveloperApi}
2626
import org.apache.spark.ml.Identifiable
27-
import org.apache.spark.sql._
27+
import org.apache.spark.sql.types.{DataType, StructField, StructType}
28+
2829

2930
/**
3031
* :: AlphaComponent ::

mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,14 @@ import org.apache.spark.mllib.regression.LinearRegressionWithSGD
2424
import org.apache.spark.sql._
2525
import org.apache.spark.storage.StorageLevel
2626

27+
2728
/**
2829
* Params for linear regression.
2930
*/
3031
private[regression] trait LinearRegressionParams extends RegressorParams
3132
with HasRegParam with HasMaxIter
3233

34+
3335
/**
3436
* :: AlphaComponent ::
3537
*

0 commit comments

Comments
 (0)