INNER CODE UNIT · Python

get_id2word

graviraja/100-Days-of-NLP · applications/classification/ner_tagging/app/app.py:24

    def get_id2word(self):
        return self.__id2word

    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)

View source record →