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