Re-add imports removed by intellij

deeplearning4j · Sep 13, 2017 · 325faec · 325faec
1 parent b732508
commit 325faec
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/...rk/dl4j-spark-ml/src/main/scala/org/deeplearning4j/spark/ml/impl/AutoEncoderWrapper.scala b/...rk/dl4j-spark-ml/src/main/scala/org/deeplearning4j/spark/ml/impl/AutoEncoderWrapper.scala
@@ -5,6 +5,7 @@ import org.apache.spark.ml.util._
 import org.apache.spark.ml.{Estimator, Model}
 import org.apache.spark.mllib.linalg.Vector
 import org.apache.spark.sql._
+import org.apache.spark.sql.expressions._
 import org.deeplearning4j.nn.conf.MultiLayerConfiguration
 import org.deeplearning4j.nn.multilayer.MultiLayerNetwork
 import org.deeplearning4j.spark.impl.multilayer.SparkDl4jMultiLayer

diff --git a/...k/dl4j-spark-ml/src/main/spark-2/scala/org/deeplearning4j/spark/ml/impl/AutoEncoder.scala b/...k/dl4j-spark-ml/src/main/spark-2/scala/org/deeplearning4j/spark/ml/impl/AutoEncoder.scala
@@ -2,13 +2,15 @@ package org.deeplearning4j.spark.ml.impl
 
 import org.apache.spark.ml.param.ParamMap
 import org.apache.spark.ml.util.Identifiable
+import org.apache.spark.ml.linalg.{Vector, Vectors}
 import org.apache.spark.sql.functions.{col, udf}
 import org.apache.spark.sql.types.StructType
 import org.apache.spark.sql.{Dataset, Row}
 import org.deeplearning4j.nn.conf.MultiLayerConfiguration
 import org.deeplearning4j.nn.multilayer.MultiLayerNetwork
 import org.deeplearning4j.spark.ml.utils._
 import org.nd4j.linalg.factory.Nd4j
+import org.apache.spark.ml.linalg.SQLDataTypes.VectorType
 
 
 class AutoEncoder(uid: String) extends AutoEncoderWrapper[AutoEncoder, AutoEncoderModel](uid){

diff --git a/...j-spark-ml/src/main/spark-2/scala/org/deeplearning4j/spark/ml/impl/SparkDl4jNetwork.scala b/...j-spark-ml/src/main/spark-2/scala/org/deeplearning4j/spark/ml/impl/SparkDl4jNetwork.scala
@@ -3,6 +3,7 @@ package org.deeplearning4j.spark.ml.impl
 import java.util
 
 import org.apache.spark.ml.param.ParamMap
+import org.apache.spark.ml.linalg.Vector
 import org.apache.spark.ml.util.Identifiable
 import org.apache.spark.mllib.linalg.Vectors
 import org.apache.spark.mllib.regression.LabeledPoint