海量数据相似度计算之simhash短文本查找

在前一篇文章 《海量数据相似度计算之simhash和海明距离》 介绍了simhash的原理,大家应该感觉到了算法的魅力。但是随着业务的增长 simhash的数据也会暴增,如果一天100w,10天就1000w了。我们如果插入一条数据就要去比较1000w次的simhash,计算量还是蛮大,普通PC 比较1000w次海明距离需要 300ms ,和5000w数据比较需要1.8 s。看起来相似度计算不是很慢,还在秒级别。给大家算一笔账就知道了:

随着业务增长需要一个小时处理100w次,一个小时为3600 *1000 = 360w毫秒,计算一下一次相似度比较最多只能消耗 360w / 100w = 3.6毫秒。300ms慢吗,慢!1.8S慢吗,太慢了!很多情况大家想的就是升级、增加机器,但有些时候光是增加机器已经解决不了问题了,就算增加机器也不是短时间能够解决的,需要考虑分布式、客户预算、问题解决的容忍时间?头大时候要相信人类的智慧是无穷的,泡杯茶,听下轻音乐:)畅想下宇宙有多大,宇宙外面还有什么东西,程序员有什么问题能够难倒呢?

加上客户还提出的几个,汇总一下技术问题:

  • 1、一个小时需要比较100w次,也就是每条数据和simhash库里的数据比较需要做到3.6毫秒。
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
Spark可以使用余弦相似度计算文本相似度。余弦相似度是一种度量两个非零向量之间的夹角的方法,它的取值范围在-1到1之间,值越接近1表示两个向量越相似,值越接近-1表示两个向量越不相似。 下面是一个使用Spark计算文本相似度的示例代码: ```python from pyspark.ml.feature import HashingTF, IDF, Tokenizer from pyspark.ml.feature import VectorAssembler from pyspark.ml.linalg import DenseVector # 创建SparkSession spark = SparkSession.builder.appName("Short Text Similarity").getOrCreate() # 加载数据集,数据集格式为id,text1,text2 data = spark.read.csv("data.csv", header=True) # 分词 tokenizer = Tokenizer(inputCol="text", outputCol="words") data = tokenizer.transform(data) # 计算TF-IDF值 hashingTF = HashingTF(inputCol="words", outputCol="rawFeatures", numFeatures=100) idf = IDF(inputCol="rawFeatures", outputCol="features") idfModel = idf.fit(hashingTF.transform(data)) data = idfModel.transform(hashingTF.transform(data)) # 合并特征向量 assembler = VectorAssembler(inputCols=["features1", "features2"], outputCol="features") data = assembler.transform(data) # 计算余弦相似度 dot_udf = udf(lambda x, y: float(x.dot(y)), DoubleType()) similarity = data.withColumn("similarity", dot_udf(col("features1"), col("features2"))) ``` 这个示例代码中使用了HashingTF和IDF来计算TF-IDF值,然后使用VectorAssembler将两个特征向量合并为一个,最后使用udf计算余弦相似度。这个示例中使用的数据集格式为id,text1,text2,你需要根据实际情况修改代码。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

菜鸟蜀黍

你的鼓励是我最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值