关闭

余弦相似度

56人阅读 评论(0) 收藏 举报
分类:

wiki链接: Cosine Similarity

这里给出公式:

/media/problem/cosine-similarity.png

给你两个相同大小的向量 A B,求出他们的余弦相似度

返回 2.0000 如果余弦相似不合法 (比如 A = [0] B = [0]).

样例

给出 A = [1, 2, 3]B = [2, 3 ,4].

返回 0.9926.

给出 A = [0]B = [0].

返回 2.0000

class Solution {
    /**
     * @param A: An integer array.
     * @param B: An integer array.
     * @return: Cosine similarity.
     */
    public double cosineSimilarity(int[] A, int[] B) {
        // write your code here
        if (A.length == 0 || B.length == 0 || A.length != B.length){
            return 2.0000;
        }
        int a = 0;
        int b = 0;
        int c = 0;
        for (int i = 0; i < A.length; i++){
            a += A[i] * A[i];
            b += B[i] * B[i];
            c += A[i] * B[i];
        }
        if (a == 0 || b == 0){
            return 2.0000;
        }
       return c/( Math.sqrt(a) * Math.sqrt(b));
    }
}

0
0

猜你在找
【套餐】Hadoop生态系统零基础入门
【套餐】嵌入式Linux C编程基础
【套餐】2017软考系统集成项目——任铄
【套餐】Android 5.x顶级视频课程——李宁
【套餐】深度学习入门视频课程——唐宇迪
【直播】广义线性模型及其应用——李科
【直播】从0到1 区块链的概念到实践
【直播】计算机视觉原理及实战——屈教授
【直播】机器学习之凸优化——马博士
【直播】机器学习&数据挖掘7周实训--韦玮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:6662次
    • 积分:933
    • 等级:
    • 排名:千里之外
    • 原创:91篇
    • 转载:0篇
    • 译文:0篇
    • 评论:2条
    最新评论