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

Method __getitem__

diffplanner/models/utils/word_vectorizer.py:64–80  ·  view source on GitHub ↗
(self, item)

Source from the content-addressed store, hash-verified

62 return len(self.word2vec)
63
64 def __getitem__(self, item):
65 word, pos = item.split('/')
66 if word in self.word2vec:
67 word_vec = self.word2vec[word]
68 vip_pos = None
69 for key, values in VIP_dict.items():
70 if word in values:
71 vip_pos = key
72 break
73 if vip_pos is not None:
74 pos_vec = self._get_pos_ohot(vip_pos)
75 else:
76 pos_vec = self._get_pos_ohot(pos)
77 else:
78 word_vec = self.word2vec['unk']
79 pos_vec = self._get_pos_ohot('OTHER')
80 return word_vec, pos_vec

Callers

nothing calls this directly

Calls 1

_get_pos_ohotMethod · 0.95

Tested by

no test coverage detected