Commit 437b7733 authored by lucas_miranda's avatar lucas_miranda
Browse files

Replaced for loop with vectorised mapping on ClusterOverlap regularization layer

parent dc539f02
Pipeline #102384 canceled with stages
in 2 minutes and 10 seconds
......@@ -321,24 +321,3 @@ def test_find_learning_rate():
test_model.build(X.shape)
deepof.model_utils.find_learning_rate(test_model, X, y)
def test_neighbor_latent_entropy():
X = np.random.normal(0, 1, [1500, 25, 6])
train_dataset = tf.data.Dataset.from_tensor_slices((X, X))
train_dataset = train_dataset.batch(256, drop_remainder=True)
test_model = deepof.models.GMVAE(number_of_components=10)
gmvaep = test_model.build(X.shape)[3]
gmvaep.fit(
train_dataset,
epochs=1,
callbacks=deepof.model_utils.neighbor_latent_entropy(
k=10,
encoding_dim=6,
samples=100,
validation_data=X,
),
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment