def test_secure_model(self): with tfe.protocol.Pond(): tf.random.set_random_seed(42) d = tf.keras.layers.Dense(1, input_shape=(10, ), use_bias=False) model = tf.keras.Sequential([d]) input = np.ones((1, 10)) output = model.predict(input) s_model = secure_model(model) s_output = s_model.private_predict(input) np.testing.assert_array_almost_equal(s_output, output, 4)
def test_secure_model_batch(self): with tfe.protocol.Pond(): tf.random.set_random_seed(42) d = tf.keras.layers.Dense(1, input_shape=(10, ), use_bias=False) model = tf.keras.Sequential([d]) x = np.ones((2, 10)) y = model.predict(x) s_model = secure_model(model, batch_size=2) s_y = s_model.private_predict(x) np.testing.assert_array_almost_equal(s_y, y, 3)