INNER CODE UNIT · Python
__getitem__
graviraja/100-Days-of-NLP · applications/classification/ner_tagging/app/app.py:27
def __getitem__(self, item):
if self.UNK in self.__word2id:
return self.__word2id.get(item, self.__word2id[self.UNK])
return self.__word2id[item]
def __len__(self):
return len(self.__word2id)
def id2word(self, idx):
return self.__id2word[idx]
with open(os.path.join('models', 'chars_vocab.pkl'), 'rb') as f:
chars_vocab = pickle.load(f)
with open(os.path.join('models', 'tags_vocab.pkl'), 'rb') as f:
tags_vocab = pickle.load(f)