Acabo de empezar a usar Word2vec y me preguntaba cómo podemos encontrar la palabra más cercana a un vector. Tengo este vector que es el vector promedio para un conjunto de vectores:

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

¿Hay una manera directa de encontrar la palabra más similar en mis datos de entrenamiento para este vector?

¿O la única solución es calcular la similitud de coseno entre este vector y los vectores de cada palabra en mis datos de entrenamiento, y luego seleccionar el más cercano?

Gracias.