def main(args):
if args.summary_vocab is None:
args.summary_vocab = args.summary + "_vocab"
if args.summary_label is None:
args.summary_label = args.summary + "_label"
assert os.path.isfile(args.summary)
assert os.path.isfile(args.summary_vocab)
assert os.path.isfile(args.summary_label)
print_args(args)
summary_dico = Dictionary.read_vocab(args.summary_vocab)
summary_data = Dictionary.index_summary(args.summary,
args.summary_label,
summary_dico,
args.summary + ".pth",
max_len=args.summary_max_length)
if __name__ == '__main__':
readme = ""
parser = argparse.ArgumentParser(description=readme)
parser.add_argument('--summary', help="summary data")
parser.add_argument('--summary_vocab', help="summary data vocab")
parser.add_argument('--summary_label', help="summary data label")
parser.add_argument('--summary_max_length',
type=int,
default=600,
help="summmary maximum length")
args = parser.parse_args()
if args.summary_vocab is None:
args.summary_vocab = args.summary + "_vocab"
if args.summary_label is None:
args.summary_label = args.summary + "_label"
assert os.path.isfile(args.summary)
assert os.path.isfile(args.summary_vocab)
assert os.path.isfile(args.summary_label)
print_args(args)
summary_dico = Dictionary.read_vocab(args.summary_vocab)
summary_data = Dictionary.index_summary(args.summary,
args.summary_label,
summary_dico,
args.summary + ".pth",
max_len=args.summary_max_length)
