[SPARK-17548] Word2VecModel.findSynonyms can spuriously reject the best match when invoked with a vector

• Type: Bug
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Affects Version/s: 1.4.1, 1.5.2, 1.6.2, 2.0.0
• Fix Version/s: 2.0.1, 2.1.0
• Component/s: MLlib

Environment

any

• Assignee: William Benton
• Reporter: William Benton
• Created: 14/Sep/16 22:32
• Updated: 17/Sep/16 11:51
• Resolved: 17/Sep/16 11:50

Description

The `findSynonyms` method in `Word2VecModel` currently rejects the best match a priori. When `findSynonyms` is invoked with a word, the best match is almost certain to be that word, but `findSynonyms` can also be invoked with a vector, which might not correspond to any of the words in the model's vocabulary. In the latter case, rejecting the best match is spurious.

Issue Links

• links to: [Github] Pull Request #15105 (willb)