fixed small error in pred_avg

JuliaTrustworthyAI · Jun 30, 2024 · c86dc25 · c86dc25
1 parent 3258618
commit c86dc25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.jl b/src/utils.jl
@@ -137,7 +137,7 @@ function empirical_frequency_binary_classification(
             #compute the empirical average and saved it in emp_avg in the j-th position
             push!(emp_avg, 1 / num_p_per_interval[j] * sum(y_binary[indices]))
             #TO DO: maybe substitute to bin_Centers?
-            push!(pred_avg, 1 / num_p_per_interval[j] * sum(class_probs[1, indices]))
+            push!(pred_avg, 1 / num_p_per_interval[j] * sum(class_probs[ indices]))
         end
     end
     #return the tuple