(self, pos)
| 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)) |
| 63 | if pos in POS_enumerator: |
| 64 | pos_vec[POS_enumerator[pos]] = 1 |
| 65 | else: |
| 66 | pos_vec[POS_enumerator['OTHER']] = 1 |
| 67 | return pos_vec |
| 68 | |
| 69 | def __len__(self): |
| 70 | return len(self.word2vec) |