We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 18e34dc commit 09848deCopy full SHA for 09848de
1 file changed
src/data/utils.jl
@@ -19,10 +19,9 @@ function train_test_split(data::CounterfactualData;test_size=0.2)
19
return train_data, test_data
20
end
21
22
-using CounterfactualExplanations.DataPreprocessing: unpack
23
function undersample(data::CounterfactualData, n_per_class::Int)
24
25
- X,y = unpack(data)
+ X,y = CounterfactualExplanations.DataPreprocessing.unpack(data)
26
n_classes, N = size(y)
27
if n_classes > 2
28
y_cls = Flux.onecold(y,1:n_classes)
0 commit comments