Python model_builderの例

コード例 #1

ファイル: hetero_nn_host.py プロジェクト: zhilangtaosha/FATE

 def _build_model(self):
     self.model = model_builder("host", self.hetero_nn_param)
     self.model.set_transfer_variable(self.transfer_variable)

コード例 #2

ファイル: hetero_nn_guest.py プロジェクト: yubo1993/FATE

 def _build_model(self):
     # return a hetero NN model with keras backend
     self.model = model_builder("guest", self.hetero_nn_param)
     self.model.set_transfer_variable(self.transfer_variable)