def __init__(self, save_dir): self.path = utils.get_vocab_path(save_dir) self.max_size = 100 self.vocab = {} if os.path.exists(self.path): self.vocab = json.load(open(self.path))
def __init__(self, model_dir, max_size): self.path = utils.get_vocab_path(model_dir) self.max_size = max_size self.vocab = {} if os.path.exists(self.path): self.vocab = json.load(open(self.path))