余弦相似度比较

#pip install scikit-learn
import numpy as np
from sklearn.metrics.pairwise import cosine_similarity
weight_chip = np.fromfile('C:\\Users\\chaojunche\\Desktop\\data',dtype=np.int8)
weight_chip = weight_chip.reshape(112,112,1)
w = np.fromfile('C:\\Users\\chaojunche\\Desktop\\data',dtype=np.int8)
w = w.reshape(112,112,1)
# for i in range(32):
#     print(weight_chip.flatten()[i],w.flatten()[i])
ret = cosine_similarity(weight_chip.reshape(1,-1), w.reshape(1,-1))
print (ret)