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

Method __getitem__

mogen/models/utils/word_vectorizer.py:72–88  ·  view source on GitHub ↗
(self, item)

Source from the content-addressed store, hash-verified

70 return len(self.word2vec)
71
72 def __getitem__(self, item):
73 word, pos = item.split('/')
74 if word in self.word2vec:
75 word_vec = self.word2vec[word]
76 vip_pos = None
77 for key, values in VIP_dict.items():
78 if word in values:
79 vip_pos = key
80 break
81 if vip_pos is not None:
82 pos_vec = self._get_pos_ohot(vip_pos)
83 else:
84 pos_vec = self._get_pos_ohot(pos)
85 else:
86 word_vec = self.word2vec['unk']
87 pos_vec = self._get_pos_ohot('OTHER')
88 return word_vec, pos_vec

Callers

nothing calls this directly

Calls 1

_get_pos_ohotMethod · 0.95

Tested by

no test coverage detected