Use cosine distance