70372381b6a03b21a8671b4bf8ec4ddfe2004522,autoencoder/baseline/doc_word2vec.py,,doc_word2vec,#,35

Before Change

def doc_word2vec(model, corpus, vocab, output, avg=True):
    doc_codes = {}
    for key in corpus:
        vec = get_doc_codes(model, corpus[key], vocab, avg)
        doc_codes[key] = vec.tolist()
    dump_json(doc_codes, output)

After Change

def doc_word2vec(model, corpus, vocab, output, avg=True):
    doc_codes = {}
    for key, bow in corpus.iteritems():
        vec = get_doc_codes(model, bow, vocab, avg)
        doc_codes[key] = vec.tolist()
    dump_json(doc_codes, output)

Instances