예제 #1
0
def main(args):
    if args.table_label is None:
        args.table_label = args.table + "_label"
    if args.table_vocab is None:
        args.table_vocab = args.table + "_vocab"

    assert os.path.isfile(args.table)
    assert os.path.isfile(args.table_label)
    assert os.path.isfile(args.table_vocab)

    print_args(args)

    table_dico = Dictionary.read_vocab(args.table_vocab)

    table_data = Dictionary.index_table(args.table, args.table_label,
                                        table_dico, args.table + ".pth")

def print_args(args):
    print("table:\t{}".format(args.table))
    print("table_label:\t{}".format(args.table_label))
    print("table_vocab:\t{}".format(args.table_vocab))


if __name__ == '__main__':
    readme = ""
    parser = argparse.ArgumentParser(description=readme)
    parser.add_argument('--table', help="table dataOld")
    parser.add_argument('--table_label', help="table label")
    parser.add_argument('--table_vocab', help="table vocab")
    args = parser.parse_args()

    if args.table_label is None:
        args.table_label = args.table + "_label"
    if args.table_vocab is None:
        args.table_vocab = args.table + "_vocab"

    assert os.path.isfile(args.table)
    assert os.path.isfile(args.table_label)
    assert os.path.isfile(args.table_vocab)

    print_args(args)

    table_dico = Dictionary.read_vocab(args.table_vocab)
    table_data = Dictionary.index_table(args.table, args.table_label,
                                        table_dico, args.table + ".pth")