stepwise_transfer_variable.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. #
  4. # Copyright 2019 The FATE Authors. All Rights Reserved.
  5. #
  6. # Licensed under the Apache License, Version 2.0 (the "License");
  7. # you may not use this file except in compliance with the License.
  8. # You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing, software
  13. # distributed under the License is distributed on an "AS IS" BASIS,
  14. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. # See the License for the specific language governing permissions and
  16. # limitations under the License.
  17. #
  18. ################################################################################
  19. #
  20. # AUTO GENERATED TRANSFER VARIABLE CLASS. DO NOT MODIFY
  21. #
  22. ################################################################################
  23. from federatedml.transfer_variable.base_transfer_variable import BaseTransferVariables
  24. # noinspection PyAttributeOutsideInit
  25. class StepwiseTransferVariable(BaseTransferVariables):
  26. def __init__(self, flowid=0):
  27. super().__init__(flowid)
  28. self.guest_data_info = self._create_variable(name='guest_data_info', src=['guest'], dst=['arbiter', 'host'])
  29. self.host_data_info = self._create_variable(name='host_data_info', src=['host'], dst=['arbiter', 'guest'])
  30. self.step_best = self._create_variable(name='step_best', src=['arbiter'], dst=['host', 'guest'])