(self, meta_root, prefix)
| 45 | |
| 46 | class WordVectorizer(object): |
| 47 | def __init__(self, meta_root, prefix): |
| 48 | vectors = np.load(pjoin(meta_root, '%s_data.npy'%prefix)) |
| 49 | words = pickle.load(open(pjoin(meta_root, '%s_words.pkl'%prefix), 'rb')) |
| 50 | word2idx = pickle.load(open(pjoin(meta_root, '%s_idx.pkl'%prefix), 'rb')) |
| 51 | self.word2vec = {w: vectors[word2idx[w]] for w in words} |
| 52 | |
| 53 | def _get_pos_ohot(self, pos): |
| 54 | pos_vec = np.zeros(len(POS_enumerator)) |