/sklearn-doc2vec

scikit-learn wrapper for gensim's doc2vec implementation

Primary LanguagePythonMIT LicenseMIT

Doc2vec for scikit-learn

This project enables you to use gensim's Doc2vec implementation as a sklearn Transformer, integrating seamlessly to pipelines.

Installing

Make sure to run the setup.sh script. It will install all required dependencies.

Running the sample

Use python sample.py to see it in action. It may take some time, but eventually it will print to the console retrieved documents matching the query, sorted by relevance.