now fixed as mucha as I can

experiments/Manifest.toml

@@ -444,7 +444,7 @@ version = "0.6.2"
 
 [[deps.CounterfactualExplanations]]
 deps = ["CSV", "CUDA", "CategoricalArrays", "ChainRulesCore", "DataFrames", "DecisionTree", "Distributions", "EvoTrees", "Flux", "LaplaceRedux", "LazyArtifacts", "LinearAlgebra", "Logging", "MLDatasets", "MLJBase", "MLJDecisionTreeInterface", "MLJModels", "MLUtils", "MultivariateStats", "NearestNeighborModels", "PackageExtensionCompat", "Parameters", "PrecompileTools", "ProgressMeter", "Random", "Serialization", "Statistics", "StatsBase", "Tables", "UUIDs", "cuDNN"]
-git-tree-sha1 = "5cd17c7c24187dc619ad28e90ec63f3e5543d62f"
+git-tree-sha1 = "67513a78f3f7037ae557e24a321e91229eb4468d"
 repo-rev = "main"
 repo-url = "https://github.com/JuliaTrustworthyAI/CounterfactualExplanations.jl"
 uuid = "2f13d31b-18db-44c1-bc43-ebaf2cff0be0"

experiments/mnist.jl

@@ -29,7 +29,7 @@ ce_measures = [CE_MEASURES..., ECCCo.distance_from_energy_cosine, ECCCo.distance
 
 # Parameter choices:
 params = (
-    n_individuals=N_IND_SPECIFIED ? N_IND : 1,
+    n_individuals=N_IND_SPECIFIED ? N_IND : 100,
     builder=default_builder(n_hidden=128, n_layers=1, activation=Flux.swish),
     𝒟x=Uniform(-1.0, 1.0),
     α=[1.0, 1.0, 1e-2],
@@ -45,7 +45,7 @@ params = (
     Λ=[0.1, 0.25, 0.25],
     Λ_Δ=[0.1, 0.1, 1.0],
     opt=Flux.Optimise.Descent(0.1),
-    reg_strength = 0.0005,
+    reg_strength = 0.0,
     ce_measures=ce_measures,
 )