/word2vec

Python interface to Google word2vec

Primary LanguageCApache License 2.0Apache-2.0

word2vec

travis-ci appveyor

Python interface to Google word2vec.

Training is done using the original C, other functionality is pure python + numpy.

Installation

I recommend the Anaconda python distribution

pip install word2vec

Wheel: Wheels packages for OS X and Windows are provided on Pypi on a best effort sense. The code is quite easy to compile so consider using: --no-use-wheel on Linux and OS X.

Linux: There is no wheel support for linux so you have to compile the C code. The only requirement is gcc. You can override the compilation flags if needed: CFLAGS='-march=corei7' pip install word2vec

Windows: Very experimental support based this win32 port

Usage

The easiest way is to look at this example: word2vec

The default functionality from word2vec is also available from the command line as:

  • word2vec
  • word2phrase
  • word2vec-distance
  • word2vec-word-analogy
  • word2vec-compute-accuracy

Experimental functionality on doc2vec can be found in this other example: doc2vec