Exemplo n.º 1
0
 def test_iterative_process_with_encoding_process(self):
   model_fn = model_examples.LinearRegression
   gather_process = encoding_utils.build_encoded_mean_process_from_model(
       model_fn, _test_encoder_fn('gather'))
   broadcast_process = (
       encoding_utils.build_encoded_broadcast_process_from_model(
           model_fn, _test_encoder_fn('simple')))
   iterative_process = optimizer_utils.build_model_delta_optimizer_process(
       model_fn=model_fn,
       model_to_client_delta_fn=DummyClientDeltaFn,
       server_optimizer_fn=lambda: tf.keras.optimizers.SGD(learning_rate=1.0),
       aggregation_process=gather_process,
       broadcast_process=broadcast_process)
   self._verify_iterative_process(iterative_process)
Exemplo n.º 2
0
 def test_mean_process_from_model(self):
     model_fn = model_examples.LinearRegression
     gather_process = encoding_utils.build_encoded_mean_process_from_model(
         model_fn, _test_encoder_fn('gather'))
     self.assertIsInstance(gather_process, measured_process.MeasuredProcess)