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)
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)