A Python implementation of word2vec which allows custom sampling strategies