K最近邻算法
概述
K最近邻算法适用于找出距离A坐标最近的几个点,可以用来做推荐系统
计算公式以及模拟
K最近邻算法有两个公式:距离公式,相似度公式(余弦)
距离公式
距离公式是:
\[\sqrt{(x_1-x_2)^2 + (y_1-y_2)^2}\]举个栗子:
你要做电影推荐系统,他有以下参数:
(a, b, c, d)
科幻片 动作片 爱情片 喜剧片
四个值是历史评分的总和
小明的打分是
(10, 5, 9, 4)
, 小红的打分是(9, 1, 2, 5)
, 小张的打分是(10, 3, 4, 5)
, 假设目前就这几个用户,小红几个小时前看了《独行月球》,小张几个小时前看了《流浪地球》,应该给小明推荐哪一部电影?
这里计算的是四维空间的距离,但是只要扩展一下就行
\[\sqrt{(x_1-x_2)^2 + (y_1-y_2)^2 + (z_1-z_2)^2 + (t_1-t_2)^2}\]代入公式,小明与小红在电影评分上的距离为:
\[\sqrt{(10-9)^2 + (5-1)^2 + (9-2)^2 + (4-5)^2} = \sqrt{67}\]非常简单!小明与小红的距离为$\sqrt{67}$
小明与小张在电影评分上的距离为:
\[\sqrt{(10-10)^2 + (5-3)^2 + (9-4)^2 + (4-5)^2} = \sqrt{30}\]由此可见,小明的电影偏好与小张比较相似,所以应该给小明推荐《流浪地球》
余弦相似度公式
余弦相似度的公式是
\[cos(\theta) = \frac{x_1x_2+y_1y_2}{\sqrt{x_1^2+y_1^2} \times {\sqrt{x_2^2+y_2^2}}}\]别急,这是纸老虎,给你举个栗子就懂了
同样是电影推荐系统的栗子,当然,为了减少计算量,我们只设二维空间的坐标值
(x, y)
科幻片 喜剧片
小明的电影偏好为
(10, 4)
, 小红的为(9, 5)
, 小张的为(10, 5)
,小红几个小时前看了《独行月球》,小明几个小时前看了《流浪地球》,请问?按照余弦相似度,该给小明推荐哪一部电影
代入公式,小明与小红的相似度为:
\[\frac{10 \times 9 + 4 \times 5}{\sqrt{10 ^ 2 + 4 ^ 2} \times {\sqrt{9^2+5^2}}} \approx 0.9919979\]余弦相似度的逻辑是:越接近1,越相似
小明与小张的相似度为:
\[\frac{10 \times 10 + 4 \times 5}{\sqrt{10 ^ 2 + 4 ^ 2} \times {\sqrt{10^2 + 5^2}}} \approx 0.996545\]因为小明与小张的相似度更接近1,所以应该给小明推荐《流浪地球》
Python代码
距离公式的代码解析
由于要使用到根号,所以需要导入math.sqrt
from math import sqrt
还要设置坐标
A_vector = (2, 3) # 实例所需
B_vector = (5, 9)
最后就是计算了
result = sqrt((A_vector[0]-B_vector[0]) ^ 2 + (A_vector[1]-B_vector[1]) ^ 2)
print('坐标{}与坐标{}的距离是{}'.format(A_vector, B_vector, result))
完整代码:
from math import sqrt
A_vector = (2, 3) # 实例所需
B_vector = (5, 9)
result = sqrt((A_vector[0]-B_vector[0]) ^ 2 + (A_vector[1]-B_vector[1]) ^ 2)
print('坐标{}与坐标{}的距离是{}'.format(A_vector, B_vector, result))
输出:
坐标(2, 3)与坐标(5, 9)的距离是1.7320508075688772
余弦相似度的代码解析
导入以及坐标设置就不说了,直接说计算过程
余弦相似度的计算是一个分数,先看分子的计算
(a_Vector[0] * b_Vector[0] + a_Vector[1] * b_Vector[1])
等同于以下数学公式
\[x_1x_2+y_1y_2\]再看分母的计算
# from math import sqrt
(sqrt(a_Vector[0] ** 2 +a_Vector[1] **2) * sqrt(b_Vector[0] ** 2 +b_Vector[1] ** 2)) # ** 等同于 ^
相当于以下数学公式
${\sqrt{x_1^2+y_1^2} \times {\sqrt{x_2^2+y_2^2}}}$
这下明白了吧,直接亮出完整代码
from math import sqrt
a_Vector = (2, 3)
b_Vector = (5, 9)
result = (a_Vector[0] * b_Vector[0] + a_Vector[1] * b_Vector[1]) / (sqrt(a_Vector[0] ** 2 +a_Vector[1] **2) * sqrt(b_Vector[0] ** 2 +b_Vector[1] ** 2))
print("∠ θ = {}".format(result))
关于K最近邻算法的介绍就到这,如果在实际工作中使用此算法,建议使用余弦相似度
公式计算,K最近邻算法的应用场景还很多,你可以用它来预测上下班高峰的的人数,预测成绩,做推荐系统……希望这篇博客对你有帮助!