(self, pos)
| 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)) |
| 55 | if pos in POS_enumerator: |
| 56 | pos_vec[POS_enumerator[pos]] = 1 |
| 57 | else: |
| 58 | pos_vec[POS_enumerator['OTHER']] = 1 |
| 59 | return pos_vec |
| 60 | |
| 61 | def __len__(self): |
| 62 | return len(self.word2vec) |