Python + numpy で cos 類似度の計算

cos_sim.py

#!/usr/bin/env python
import numpy as np

def cos_sim(v1, v2):
    return np.dot(v1, v2) / (np.linalg.norm(v1) * np.linalg.norm(v2))

if __name__ == '__main__':
    x = np.array([1, 1, 1, 1, 1])
    y = np.array([1, 0, 1, 0, 1])
    z = np.array([0, 1, 0, 0, 0])
    
    print cos_sim(x, y)
    print cos_sim(y, z)
    print cos_sim(z, x)