Since the TSS is positioned in the same vector space with original words, we could utilize the embedding cosine similarity between TSS and given words to determine their TSS.
we tag a given word with the TSS, if the cosine similarity is above a predefined threshold score 5.