improvement of scala spark code snippet for calculating cosine-similarity in terms of execution time

Question

I have a dataframe that is constructed using the transform() method of the VectorAssembler class. Besides, I have a trained k-means model that can output the i-th center point w