使用Scala实现VSM向量空间模型算法实现的简单实例

DataWorker_Q

于 2020-04-01 19:05:20 发布

阅读量316

点赞数

分类专栏： Scala 文章标签： scala

本文链接：https://blog.youkuaiyun.com/qq_32834005/article/details/105253088

版权

Scala 专栏收录该内容

2 篇文章

订阅专栏

本文深入探讨了VSM(Vector Space Model)向量空间模型算法，该算法广泛应用于文档排名和推荐系统中。通过计算两个向量的余弦相似度，VSM能够衡量文档之间的相似性，为信息检索和文本挖掘提供了有力的支持。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

VSM (Vector Space Model) 向量空间模型算法，可用于文档排名，推荐系统

def main(args: Array[String]): Unit = {
    val a1= Array(1,2,5)
    val a2= Array(3,4,6)
    val r = vcos(a1, a2)
    println(r)
      
  }
  
  /*取模*/
  def vmod(a:Array[Int]):Double={
    return math.sqrt{a.map{x => x*x}.sum}
  }
  
  /*两个向量的内积和*/
  def vmulti(x:Array[Int],y:Array[Int])={
    val multiArr = ArrayBuffer[Double]()
    for(i <- 0 until x.length; j <- 0 until y.length; if i==j){
      if(i == j) multiArr.append((x(i)*y(i)).toDouble)
    }
    multiArr.sum
  }
  
  /*余弦*/
  def vcos(x:Array[Int],y:Array[Int])={
    vmulti(x, y)/(vmod(x)*vmod(y))
  }