INNER CODE UNIT · Python

__init__

graviraja/100-Days-of-NLP · applications/generation/image_captioning/latex_app/app.py:30

    def __init__(self):
        self.sign2id = {
            "<s>": START_TOKEN,
            "</s>": END_TOKEN,
            "<pad>": PAD_TOKEN,
            "<unk>": UNK_TOKEN
        }

        self.id2sign = dict((id, sign) for sign, id in self.sign2id.items())
        self.length = 4
    
    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):

View source record →