Merge pull request #18 from JuliaTrustworthyAI/17-getting-and-error-f…

…or-dice

17 getting and error for dice

Project.toml

@@ -1,7 +1,7 @@
 name = "TaijaPlotting"
 uuid = "bd7198b4-c7d6-400c-9bab-9a24614b0240"
 authors = ["Patrick Altmeyer"]
-version = "1.0.8"
+version = "1.0.9"
 
 [deps]
 CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"

src/CounterfactualExplations/data.jl

@@ -1,3 +1,5 @@
+using MLUtils
+
 function embed(data::CounterfactualData, X::AbstractArray = nothing; dim_red::Symbol = :pca)
 
     # Training compressor:
@@ -20,9 +22,9 @@ function embed(data::CounterfactualData, X::AbstractArray = nothing; dim_red::Sy
     end
 
     # Transforming:
-    X = typeof(X) <: Vector{<:Matrix} ? hcat(X...) : X
+    X = typeof(X) <: Vector{<:Matrix} ? MLUtils.stack(X, dims=2) : X
     if !isnothing(tfn) && !isnothing(X)
-        X = MultivariateStats.predict(tfn, X)
+        X = mapslices(x -> MultivariateStats.predict(tfn, x), X, dims = 1)
     else
         X = isnothing(X) ? X_train : X
     end

test/CounterfactualExplanations.jl

@@ -24,6 +24,15 @@ using TaijaData
         plot(ce)
         plot(ce; plot_proba = true, zoom = -0.1f32)
         TaijaPlotting.animate_path(ce)
+
+        @test true
+
+        @testset "Multiple counterfactuals" begin
+            ce = generate_counterfactual(x, target, counterfactual_data, M, generator; num_counterfactuals = 5)
+            plot(ce)
+            @test true
+        end
+
     end
 
     @testset "Multi-dim" begin
@@ -43,6 +52,8 @@ using TaijaData
         plot(M, counterfactual_data)
         plot(ce)
         TaijaPlotting.animate_path(ce)
+
+        @test true
     end
 
     @test true