def _get_model_meta(self):
model_meta = HeteroNNMeta()
model_meta.batch_size = self.batch_size
model_meta.hetero_nn_model_meta.CopyFrom(
self.model.get_hetero_nn_model_meta())
return model_meta
def _get_model_meta(self):
model_meta = HeteroNNMeta()
model_meta.task_type = self.task_type
model_meta.module = 'HeteroNN'
model_meta.batch_size = self.batch_size
model_meta.epochs = self.epochs
model_meta.early_stop = self.early_stop
model_meta.tol = self.tol
# model_meta.interactive_layer_lr = self.hetero_nn_param.interacitve_layer_lr
model_meta.hetero_nn_model_meta.CopyFrom(self.model.get_hetero_nn_model_meta())
return model_meta