Exemplo n.º 1
0
def main():
    configure_tensorflow()
    args = parse_args()

    if args.format_data:
        format_data()
    if args.train:
        train(args.batch_size, args.epochs, args.verbose, args.data_percentage, args.patience)
    if args.plot:
        plot()
    if args.out:
        output(args.file_count)
Exemplo n.º 2
0
def main():
    args = parse_args()

    if args.format_data:
        format_data(args.file_count)
    else:
        configure_tensorflow()

    if args.train:
        train(args.batch_size, args.epochs, args.verbose, args.data_percentage,
              args.patience)
    if args.encode:
        encode(args.batch_size)
    if args.generate:
        generate(args.file_count, args.batch_size)
    if args.cross:
        cross(args.file_count, args.batch_size)
    if args.plot:
        plot(args.batch_size)
    if args.gui:
        gui()
    if args.out:
        output(args.file_count, args.batch_size)