INNER CODE UNIT · Python

id2word

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

    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)

with open(os.path.join('models', 'words_vocab.pkl'), 'rb') as f:
    words_vocab = pickle.load(f)


@st.cache
def load_model():
    vocab_size = len(words_vocab)
    sent_pad_token = words_vocab[words_vocab.PAD]

View source record →