next bug

Author: pat-alt

docs/src/paper/experiments/_real_world.qmd

@@ -15,7 +15,7 @@ max_obs = 2500
 data_sets = AlgorithmicRecourseDynamics.Data.load_real_world(max_obs; data_dir=data_path)
 choices = [
     :cal_housing, 
-    :credit_default, 
+    # :credit_default, 
     :gmsc, 
 ]
 data_sets = filter(p -> p[1] in choices, data_sets)
@@ -40,10 +40,10 @@ models = [
     :FluxEnsemble
 ]
 generators = Dict(
-    # :Greedy=>GreedyGenerator(), 
-    # :Generic=>GenericGenerator(),
+    :Greedy=>GreedyGenerator(), 
+    :Generic=>GenericGenerator(),
     :REVISE=>REVISEGenerator(),
-    # :DICE=>DiCEGenerator(),
+    :DICE=>DiCEGenerator(),
 )
 ```
 

src/experiments/functions.jl

@@ -111,13 +111,14 @@ function choose_individuals(experiment::Experiment, recourse_systems::AbstractAr
     args = experiment.fixed_parameters
     target, μ = experiment.target, args.μ
 
-    candidates = map(recourse_systems) do x
-        n_classes = size(x.data.y, 1)
+    candidates = map(recourse_systems) do sys
+        ŷ = probs(sys.model, sys.data.X)
+        n_classes = size(sys.data.y, 1)
         if n_classes == 1
-            cand_ = findall(vec(x.data.y) .!= target)
+            cand_ = findall(vec(ŷ) .!= target)
         else
-            y = Flux.onecold(x.data.y, 1:n_classes)
-            cand_ = findall(vec(y) .!= target)
+            ŷ = Flux.onecold(ŷ, 1:n_classes)
+            cand_ = findall(vec(ŷ) .!= target)
         end
         return cand_
     end