(datasets, output_dir)
dump_pkl
dump_pkl
dump_pkl
def dump_pkl(datasets, output_dir):
    """
        dump_pkl
    """
    chunk_size = parse_args.chunk_size
    for i in range(10):
        sub_dataset = datasets[i * chunk_size:(i + 1) * chunk_size, :]
        sub_dataset.dump(output_dir + "/" + 'data_batch_' + str(i) + '.pkl')
def dump_pkl ( datasets , output_dir ) : chunk_size = parse_args . chunk_size for i in range ( 10 ) : sub_dataset = datasets [ i * chunk_size : ( i + 1 ) * chunk_size , : ] sub_dataset . dump ( output_dir + "/" + 'data_batch_' + str ( i ) + '.pkl' )