2573c649518391ada6214cfc72d20421dfac4072,src/preprocess.py,,get_embeddings,#Any#Any#Any#,194

Before Change

        try:
            assert word in word2vec
        except AssertionError as error:
            log.debug(error)
            pdb.set_trace()
        embeddings[idx] = word2vec[word]
    embeddings[vocab.get_token_index("@@PADDING@@")] = 0.
    embeddings = torch.FloatTensor(embeddings)

After Change

            idx = vocab.get_token_index(word)
            if idx != unk_idx:
                idx = vocab.get_token_index(word)
                embeddings[idx] = np.array(list(map(float, vec.split())))
    embeddings[vocab.get_token_index("@@PADDING@@")] = 0.
    embeddings = torch.FloatTensor(embeddings)
    log.info("\tFinished loading embeddings")
    return embeddings

Instances