Добрый день, спасибо за интерес к статье. Вообще, данный эмбеддер позиционируется как векторизатор для предложений. На сколько корректно он будет работать в контексте сравнения двух слов/понятий сказать не могу. Однако вот результат сравнения, который вы просите:
щенок
0.5037827
сукин сын
щенок
0.4377274
сын
щенок
0.7690866
собачка
Как ML помогает при аудите качества клиентского сервиса