INNER CODE UNIT · Python
add_sign
graviraja/100-Days-of-NLP · applications/generation/image_captioning/latex_app/app.py:41
def add_sign(self, sign):
if sign not in self.sign2id:
self.sign2id[sign] = self.length
self.id2sign[self.length] = sign
self.length += 1
def __len__(self):
return self.length
def __call__(self, sign):
if not sign in self.sign2id:
return self.sign2id['<unk>']
return self.sign2id[sign]
def load_vocab():
with open(os.path.join("model", 'vocab.pkl'), 'rb') as f:
vocab = pickle.load(f)
return vocab