scala - Scala 中的海灵格距离

Question

给定 2 Array[Double]其中包含离散概率分布，我如何实现 Hellinger Distance在斯卡拉？

def hellingerSimilarity(a1: Array[Double], a2: Array[Double]): Double = {


}

我正在尝试使用scala.math.sqrt函数，但它只适用于单个 Double而不是Array 。我习惯了Python的pairwise metrics ，Scala/Spark中有类似的库吗？

Answer 1

下面只是scala代码中的数学公式:

def hellingerSimilarity(a1: Array[Double], a2: Array[Double]): Double = {
  val tupleSum = a1.zip(a2).map(x => pow(sqrt(x._1) - sqrt(x._2),2)).reduce(_+_)
  (1/sqrt(2)) * sqrt(tupleSum)
}