Acabei de começar a usar o Word2vec e fiquei imaginando como podemos encontrar a palavra mais próxima de um vetor. Eu tenho esse vetor que é o vetor médio para um conjunto de vetores:

array([-0.00449447, -0.00310097, 0.02421786, ...], dtype=float32)

Existe uma maneira direta de encontrar a palavra mais semelhante nos meus dados de treinamento para esse vetor?

Ou a única solução é calcular a semelhança de cosseno entre esse vetor e os vetores de cada palavra nos meus dados de treinamento e selecionar o mais próximo?

Obrigado.