在scala中计算余弦相似度

import org.apache.commons.math3.linear.{ArrayRealVector, RealVector}
import org.apache.commons.math3.linear._

object CosineSimilarityCalculator {
  def calculateSimilarity(vectorA: Array[Double], vectorB: Array[Double]): Double = {
    val realVectorA: RealVector = new ArrayRealVector(vectorA, false)
    val realVectorB: RealVector = new ArrayRealVector(vectorB, false)
    
    val dotProduct: Double = realVectorA.dotProduct(realVectorB)
    val normA: Double = realVectorA.getNorm
    val normB: Double = realVectorB.getNorm
    
    dotProduct / (normA * normB)
  }
}

object Main {
  def main(args: Array[String]): Unit = {
    val vectorA = Array(1.0, 2.0, 3.0)
    val vectorB = Array(4.0, 5.0, 6.0)
    
    val similarity = CosineSimilarityCalculator.calculateSimilarity(vectorA, vectorB)
    println("Cosine similarity: " + similarity)
  }
}