mc_func.py 文件源码

def forward_parallel(forward_process,X,n,n_jobs,extra_par): 

    n_row=X.shape[0]

    parallelizer = Parallel(n_jobs=n_jobs)

    tasks_iterator = ( delayed(forward_process)(X_row,n,extra_par) 
                      for X_row in np.split(X,n_row))

    result = parallelizer( tasks_iterator )
    # Merging the output of the jobs
    return np.vstack(result)