now fixed

JuliaTrustworthyAI · Nov 8, 2024 · 5597c7a · 5597c7a
1 parent 295b4d0
commit 5597c7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/data_preprocessing/utils.jl b/src/data_preprocessing/utils.jl
@@ -45,7 +45,7 @@ function train_test_split(data::CounterfactualData; test_size=0.2, keep_class_ra
         class_samples = [
             sample(
                 findall(vec(y .== cls)),
-                Int(floor(test_size * cls_ratio * n_per_class * 10));
+                Int(floor(test_size * cls_ratio * N));
                 replace=false,
             ) for (cls, cls_ratio) in zip(classes_, class_ratios)
         ]