File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
mllib/src/main/scala/org/apache/spark/ml/feature Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -55,7 +55,7 @@ class VectorAssembler extends Transformer with HasInputCols with HasOutputCol {
5555 schema(c).dataType match {
5656 case DoubleType => UnresolvedAttribute (c)
5757 case t if t.isInstanceOf [VectorUDT ] => UnresolvedAttribute (c)
58- case _ : NumericType | StringType =>
58+ case _ : NumericType =>
5959 Alias (Cast (UnresolvedAttribute (c), DoubleType ), s " ${c}_double_ $uid" )()
6060 }
6161 }
@@ -68,7 +68,7 @@ class VectorAssembler extends Transformer with HasInputCols with HasOutputCol {
6868 val outputColName = map(outputCol)
6969 val inputDataTypes = inputColNames.map(name => schema(name).dataType)
7070 inputDataTypes.foreach {
71- case _ : NumericType | StringType =>
71+ case _ : NumericType =>
7272 case t if t.isInstanceOf [VectorUDT ] =>
7373 case other =>
7474 throw new IllegalArgumentException (s " Data type $other is not supported. " )
You can’t perform that action at this time.
0 commit comments