Post rebuttal

Created by: pat-alt

Streamlines the code base for revision and further experiments.

experiments/mnist.jl

@@ -18,6 +18,10 @@ counterfactual_data.generative_model = vae
 test_data = load_mnist_test()
 
 # Models:
+builder = MLJFlux.@builder Flux.Chain(
+    Dense(n_in, n_hidden, activation),
+    Dense(n_hidden, n_out),
+)
 
 # Generators:
 eccco_generator = ECCCoGenerator(