0%

CS231N-assignment1-无循环的距离公式推导

发表于 更新于 分类于 阅读次数：
本文字数： 418 阅读时长 ≈ 1 分钟

2019年5月29日 下午9:39

无循环的距离公式推导：两种表示

  1. 【实验小结】cs231n assignment1 knn 部分 - zhyh1435589631的专栏 - CSDN博客
  2. 欢迎来到实力至上主义的CS231n教室（一） 机器学习基础 - 知乎

python 代码

  1. 先搭建起整个1-3步的框架
    1
    dists = np.sqrt(self.getNormMatrix(X, num_train).T + self.getNormMatrix(self.X_train, num_test) - 2 * np.dot(X, self.X_train.T))
  2. 将第一(二)步模块化：这里在np.sum的时候也使用了向量化的思想
    1
    2
    3
    4
    5
    def getNormMatrix(self, x, lines_num):
        """
        Get a lines_num x size(x, 1) matrix
        """ 
        return np.ones((lines_num, 1)) * np.sum(np.square(x), axis = 1)