关键词匹配优化(第1篇)—— 测试计算过程

昨天大致把思路理清楚了,用one-hot的方式把关键词按字拆开编码,今天尝试可行性。

目前主流的文本向量化方式主要包括one-hot、tf (term-frequency)和tf-idf (term frequency–inverse document frequency)这三种,越往后准确度应该越高。

我的需求比较简单,只有一百多个相对较短的关键词,所以先用one-hot方式向量化,把整个流程跑通,再回来测试准确度的区别。

分词的话可以用python的jieba包,因为词语比较短,所以也先不分词了,按字拆分。

一百多个词拆下来一共200多个关键字,感觉也还ok。

思路的验证是在Excel中做的,毕竟数据量小,看起来也方便。

流程是:先用vba把词拆成字(用Excel公式也可以完成);用vba的字典功能把字去重放到首行;然后用countif公式得出one-hot编码。

关键词编码

后面遇到新词之后计算新词的编码,对比两个向量就可以计算相似度了。

向量的相似度用 余弦相似度 方法,不赘述,具体解释可以百度,公式如下。

余弦相似度公式

找了一段用numpy计算余弦相似度的代码,进行测试。

import numpy as np

def cos_sim(vector_a, vector_b):
    """
    计算两个向量之间的余弦相似度
    :param vector_a: 向量 a 
    :param vector_b: 向量 b
    :return: sim
    """
    vector_a = np.mat(vector_a)
    vector_b = np.mat(vector_b)
    num = float(vector_a * vector_b.T)
    denom = np.linalg.norm(vector_a) * np.linalg.norm(vector_b)
    sim = num / denom
    return sim

print( cos_sim(np.array([[1, 1, 1,1,1,1,1,1]]),np.array([[1, 2, 1,1,1,1,1,1]])) )

成功计算出这两个测试向量的相似度

目前整个计算流程基本算是跑通了,接下来就是用C#实现以上功能。

posted @ 2020-04-12 12:49  何未生  阅读(539)  评论(0编辑  收藏  举报