diff --git a/experiments/Manifest.toml b/experiments/Manifest.toml index ffae61c4110a83b55e03f465e20e50a02af2a1cb..0b44ff76256620e2a966c04c760240189ad8d063 100644 --- a/experiments/Manifest.toml +++ b/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" diff --git a/experiments/mnist.jl b/experiments/mnist.jl index 03e294fcaa0f6d66e7d756dd918595c00976708f..e48bbcfba6fb556823b93502f569b1dd6cdbfc28 100644 --- a/experiments/mnist.jl +++ b/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, )