feat_test.py 文件源码

def getDist(feat1, feat2, metric):
    pair_num = len(feat1)
    import sklearn.metrics.pairwise as pw
    mt = pw.pairwise_distances(feat1, feat2, metric=metric)
    distance = np.empty((pair_num,))
    for i in xrange(pair_num):
        distance[i] = mt[i,i]
    return distance

# Extract feature via network.