schoennenbeck / swem

pytorch implementation of the simple word embedding model.

Simple word embedding models

A pytorch implementation of the Simple Word Embedding Model from the paper Baselines need more love and some additional models and utilities.

Installation

From pypi

To install this package from pypi simply run

pip install pytorch-swem

From source

To install from source clone this repository and install via pip:

git clone https://github.com/schoennenbeck/swem.git
cd swem
pip install .

Usage

For an example of how to use this package please see the demo-notebook .

Docs

The documentation is hosted on readthedocs.