Python HeteroNNParam Examples

Example #1

File: hetero_nn_host.py Project: UnreliableBuilder/Fate

    def _get_model_param(self):
        model_param = HeteroNNParam()
        model_param.hetero_nn_model_param.CopyFrom(
            self.model.get_hetero_nn_model_param())
        model_param.best_iteration = -1 if self.validation_strategy is None else self.validation_strategy.best_iteration

        return model_param

Example #2

File: hetero_nn_host.py Project: yubo1993/FATE

    def _get_model_param(self):
        model_param = HeteroNNParam()
        model_param.iter_epoch = self.iter_epoch
        model_param.header.extend(self._header)
        model_param.hetero_nn_model_param.CopyFrom(self.model.get_hetero_nn_model_param())
        model_param.best_iteration = self.callback_variables.best_iteration

        return model_param

Example #3

File: hetero_nn_guest.py Project: errord/FATE-1

    def _get_model_param(self):
        model_param = HeteroNNParam()
        model_param.iter_epoch = self.iter_epoch
        model_param.hetero_nn_model_param.CopyFrom(self.model.get_hetero_nn_model_param())
        model_param.num_label = self.num_label

        for loss in self.history_loss:
            model_param.history_loss.append(loss)

        return model_param

Example #4

File: hetero_nn_guest.py Project: UnreliableBuilder/Fate

    def _get_model_param(self):
        model_param = HeteroNNParam()
        model_param.iter_epoch = self.iter_epoch
        model_param.hetero_nn_model_param.CopyFrom(self.model.get_hetero_nn_model_param())
        model_param.num_label = self.num_label
        model_param.best_iteration = -1 if self.validation_strategy is None else self.validation_strategy.best_iteration

        for loss in self.history_loss:
            model_param.history_loss.append(loss)

        return model_param

Example #5

File: hetero_nn_host.py Project: zhilangtaosha/FATE

    def _get_model_param(self):
        model_param = HeteroNNParam()
        model_param.hetero_nn_model_param.CopyFrom(
            self.model.get_hetero_nn_model_param())

        return model_param