def run(self, train_X, train_y):
# tell slave that it should do hyper-parameter search
self._task_desc[0] = 0
self._task_desc[1] = MPI_MSG_CV
comm.Bcast([self._task_desc, MPI.INT], root=0)
comm.bcast((train_X, train_y), root=0)
self._data_X = train_X
self._data_y = train_y
root_result_batch = self._scatter_work()
return self._gather_work(root_result_batch)
评论列表
文章目录