import numpy as np
... 생략
cos = util.pytorch_cos_sim(embedding_a, embedding_b)[0]
doc_cohesion.append(cos.item())
cos_sim = np.mean(doc_cohesion)
이 상황에서 cosine similarity의 값은 tensor type
으로 떨어진다.
그래서 cos.item()
을 활용하여 float
형으로 떨어지도록 해야함.
출처 : https://discuss.pytorch.org/t/fastest-way-to-convert-tensor-to-float/21282