Python _preprocess_dummy_batch Exemples

Exemple #1

Fichier : keras_utils_test.py Projet : tyoungroy/federated

 def test_preprocess_batch_non_mapping(self):
   tuple_dummy_batch = ([[1.0], [2.0]], [3.0, 4.0])
   processed_batch = keras_utils._preprocess_dummy_batch(tuple_dummy_batch)
   self.assertIsInstance(processed_batch, tuple)
   # Assert that the Python float was converted to a tf.Tensor.
   self.assertIsInstance(processed_batch[0][0][0], tf.Tensor)
   self.assertAllClose(processed_batch, tuple_dummy_batch)

Exemple #2

 def test_preprocess_batch_only_converts_leaves_to_tensors(self):
     dummy_batch = collections.OrderedDict([
         ('x', [[np.zeros([1, 1], dtype=np.float32)],
                [np.zeros([1, 1], dtype=np.float32)]]),
         ('y', [np.zeros([1, 1], dtype=np.float32)]),
     ])
     processed_batch = keras_utils._preprocess_dummy_batch(dummy_batch)
     self.assertIsInstance(processed_batch['x'], list)
     self.assertIsInstance(processed_batch['y'], list)
     self.assertIsInstance(processed_batch['x'][0], list)
     self.assertIsInstance(processed_batch['x'][0][0], tf.Tensor)
     self.assertIsInstance(processed_batch['y'][0], tf.Tensor)