def test_sparse_variational_optimize_with_defaults() -> None:
x_observed = np.linspace(0, 100, 100).reshape((-1, 1))
y_observed = _3x_plus_gaussian_noise(x_observed)
data = x_observed, y_observed
dataset = Dataset(*data)
optimizer = create_optimizer(tf.optimizers.Adam(), dict(max_iter=20))
model = SparseVariational(svgp_model(x_observed, y_observed), optimizer=optimizer)
loss = model.model.training_loss(data)
model.optimize(dataset)
assert model.model.training_loss(data) < loss
def test_sparse_variational_optimize(batcher: DatasetTransformer, compile: bool) -> None:
x_observed = np.linspace(0, 100, 100).reshape((-1, 1))
y_observed = _3x_plus_gaussian_noise(x_observed)
data = x_observed, y_observed
dataset = Dataset(*data)
optimizer = create_optimizer(
tf.optimizers.Adam(),
dict(max_iter=10, batch_size=10, dataset_builder=batcher, compile=compile),
)
model = SparseVariational(svgp_model(x_observed, y_observed), optimizer=optimizer)
loss = model.model.training_loss(data)
model.optimize(dataset)
assert model.model.training_loss(data) < loss