MCPcopy Create free account
hub / github.com/MotrixLab/FineMoGen / __init__

Method __init__

mogen/models/utils/word_vectorizer.py:53–59  ·  view source on GitHub ↗
(self, meta_root, prefix)

Source from the content-addressed store, hash-verified

51class WordVectorizer(object):
52
53 def __init__(self, meta_root, prefix):
54 vectors = np.load(pjoin(meta_root, '%s_data.npy' % prefix))
55 words = pickle.load(
56 open(pjoin(meta_root, '%s_words.pkl' % prefix), 'rb'))
57 word2idx = pickle.load(
58 open(pjoin(meta_root, '%s_idx.pkl' % prefix), 'rb'))
59 self.word2vec = {w: vectors[word2idx[w]] for w in words}
60
61 def _get_pos_ohot(self, pos):
62 pos_vec = np.zeros(len(POS_enumerator))

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected