当我调用一个函数时,它可以工作。但是当我在UDF中调用该函数时将无效。
这是完整的代码。
val sparkConf = new SparkConf().setAppName("HiveFromSpark").set("spark.driver.allowMultipleContexts","true")
val sc = new SparkContext(sparkConf)
val hive = new org.apache.spark.sql.hive.HiveContext(sc)
///////////// UDFS
def toDoubleArrayFun(vec:Any) : scala.Array[Double] = {
return vec.asInstanceOf[WrappedArray[Double]].toArray
}
def toDoubleArray=udf((vec:Any) => toDoubleArrayFun(vec))
//////////// PROCESS
var df = hive.sql("select vec from mst_wordvector_tapi_128dim where word='soccer'")
println("==== test get value then transform")
println(df.head().get(0))
println(toDoubleArrayFun(df.head().get(0)))
println("==== test transform by udf")
df.withColumn("word_v", toDoubleArray(col("vec")))
.show(10);
然后是输出。
sc: org.apache.spark.SparkContext = org.apache.spark.SparkContext@6e9484ad
hive: org.apache.spark.sql.hive.HiveContext =
toDoubleArrayFun: (vec: Any)Array[Double]
toDoubleArray: org.apache.spark.sql.UserDefinedFunction
df: org.apache.spark.sql.DataFrame = [vec: array<double>]
==== test get value then transform
WrappedArray(-0.88675,, 0.0216657)
[D@4afcc447
==== test transform by udf
org.apache.spark.SparkException: Job aborted due to stage failure: Task 0 in stage 2.0 failed 4 times, most recent failure: Lost task 0.3 in stage 2.0 (TID 5, xdad008.band.nhnsystem.com): java.lang.ClassNotFoundException: $iwC$$iwC$$iwC$$iwC$$iwC$$$$5ba2a895f25683dd48fe725fd825a71$$$$$$iwC$$anonfun$toDoubleArray$1
at java.net.URLClassLoader.findClass(URLClassLoader.java:381)
at java.lang.ClassLoader.loadClass(ClassLoader.java:424)
at java.lang.ClassLoader.loadClass(ClassLoader.java:357)
此处为完整输出。 https://gist.github.com/jeesim2/efb52f12d6cd4c1b255fd0c917411370
正如你所看到的那样,&#34; toDoubleArrayFun&#34;函数运行良好,但在udf中它声明ClassNotFoundException。
我无法更改配置单元数据结构,需要将vec转换为Array [Double]以生成Vector实例。
那么上面的代码有什么问题?
Spark版本是1.6.1
更新1
Hive table&#39; vec&#39;列类型为&#34; array<double>
&#34;
下面的代码也会导致错误
var df = hive.sql("select vec from mst_wordvector_tapi_128dim where
word='hh'")
df.printSchema()
var word_vec = df.head().get(0)
println(word_vec)
println(Vectors.dense(word_vec))
输出
df: org.apache.spark.sql.DataFrame = [vec: array<double>]
root
|-- vec: array (nullable = true)
| |-- element: double (containsNull = true)
==== test get value then transform
word_vec: Any = WrappedArray(-0.88675,...7)
<console>:288: error: overloaded method value dense with alternatives:
(values: Array[Double])org.apache.spark.mllib.linalg.Vector <and>
(firstValue: Double,otherValues:Double*)org.apache.spark.mllib.linalg.Vector
cannot be applied to (Any)
println(Vectors.dense(word_vec))
这意味着hive&#39; array<double>
&#39;列不能转换为Array<Double>
实际上我想计算距离:用两个array<double>
列加倍。
如何根据array<double>
列添加Vector列?
典型方法是
Vectors.sqrt(Vectors.dense(Array<Double>, Array<Double>)
答案 0 :(得分:2)
由于udf
函数必须进行序列化和反序列化过程,any
DataType将不起作用。您必须定义要传递给udf
函数的列的确切数据类型。
根据您问题中的输出,您的数据框中只有一列,即vec
类型为Array[Double]
df: org.apache.spark.sql.DataFrame = [vec: array<double>]
实际上不需要udf函数,因为vec
列已经是Array
dataType,这也是udf
函数正在执行的操作,即将值转换为{ {1}}。
现在,您的其他函数调用正在运行
Array[Double]
因为不需要序列化和反序列化过程,所以只需要 scala 函数调用。