Skip to main content

Context Encoders (ConEc) as an extension of word2vec

Project description

Context Encoders (ConEc)
========================

With this code you can train and evaluate Context Encoders (ConEc), an extension of word2vec, which can learn word embeddings from large corpora and create out-of-vocabulary embeddings on the spot as well as distinguish between multiple meanings of words based on their local contexts.
For further details on the model and experiments please refer to the paper_ - and of course if any of this code was helpful for your research, please consider citing it: ::
@inproceedings{horn2017conecRepL4NLP,
author = {Horn, Franziska},
title = {Context encoders as a simple but powerful extension of word2vec},
booktitle = {Proceedings of the 2nd Workshop on Representation Learning for NLP},
year = {2017},
organization = {Association for Computational Linguistics},
pages = {10--14}
}

.. _paper: https://arxiv.org/abs/1706.02496

The code is intended for research purposes. It was programmed for Python 2.7, but should theoretically also run on newer Python 3 versions - no guarantees on this though (open an issue if you find a bug, please)!

installation
------------
You either download the code from here and include the conec folder in your ``$PYTHONPATH`` or install (the library components only) via pip:

``$ pip install conec``

conec library components
------------------------

dependencies: numpy, scipy

- ``word2vec.py``: code to train a standard word2vec model, adapted from the corresponding gensim_ implementation.
- ``context2vec.py``: code to build a sparse context matrix from a large collection of texts; this context matrix can then be multiplied with the corresponding word2vec embeddings to give the context encoder embeddings:

.. code-block:: python

# get the text for training
sentences = Text8Corpus('data/text8')
# train the word2vec model
w2v_model = word2vec.Word2Vec(sentences, mtype='cbow', hs=0, neg=13, embed_dim=200, seed=3)
# get the global context matrix for the text
context_model = context2vec.ContextModel(sentences, min_count=w2v_model.min_count, window=w2v_model.window, wordlist=w2v_model.index2word)
context_mat = context_model.get_context_matrix(fill_diag=False, norm='max')
# multiply the context matrix with the (length normalized) word2vec embeddings
# to get the context encoder (ConEc) embeddings
conec_emb = context_mat.dot(w2v_model.syn0norm)
# renormalize so the word embeddings have unit length again
conec_emb = conec_emb / np.array([np.linalg.norm(conec_emb, axis=1)]).T

.. _gensim: https://radimrehurek.com/gensim/


examples
--------
additional dependencies: sklearn, unidecode

``test_analogy.py`` and ``test_ner.py`` contain the code to replicate the analogy and named entity recognition (NER) experiments discussed in the aforementioned paper.

To run the analogy experiment, it is assumed that the `text8 corpus`_ or `1-billion corpus`_ as well as the `analogy questions`_ are in a data directory.

To run the named entity recognition experiment, it is assumed that the corresponding `training and test files`_ are located in the data/conll2003 directory.

.. _`text8 corpus`: http://mattmahoney.net/dc/text8.zip
.. _`1-billion corpus`: http://code.google.com/p/1-billion-word-language-modeling-benchmark/
.. _`analogy questions`: https://code.google.com/archive/p/word2vec/
.. _`training and test files`: http://www.cnts.ua.ac.be/conll2003/ner/


If you have any questions please don't hesitate to send me an `email <mailto:cod3licious@gmail.com>`_ and of course if you should find any bugs or want to contribute other improvements, pull requests are very welcome!


Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
conec-1.0.1-py2.py3-none-any.whl (14.1 kB) Copy SHA256 hash SHA256 Wheel py2.py3
conec-1.0.1.tar.gz (14.1 kB) Copy SHA256 hash SHA256 Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page