Commit cc080f19 authored by Philippe Veber's avatar Philippe Veber
Browse files

Simulation_pipeline: fixed PR AUC calculation in Mutsel.benchmark

parent 4d3d17c9
......@@ -184,12 +184,12 @@ module Mutsel = struct
let ancestral_counts = collect_counts `Ancestral in
let convergent_counts = collect_counts `Convergent in
let average_precision =
let oracle = Array.init nsites ~f:(fun i -> i < n_h0) in
let oracle = Array.init nsites ~f:(( <= ) n_h0) in
let module PR = Prc.Precision_recall in
List.map results ~f:(fun (_, scores) ->
let Prc.Dataset xs as data = U.make_prc_dataset scores oracle in
let n = List.count xs ~f:snd in
let theta_hat = PR.auc_trapezoidal_lt data in
let theta_hat = PR.auc_average_precision data in
let lb, ub = PR.logit_confidence_interval ~alpha:0.05 ~theta_hat ~n in
theta_hat, (lb, ub)
)
......@@ -200,7 +200,7 @@ module Mutsel = struct
average_precision ; profiles ; n_h0 ; n_ha
}
in
Workflow.plugin ~version:3 ~descr:"simulation_pipeline.benchmark" f
Workflow.plugin ~version:4 ~descr:"simulation_pipeline.benchmark" f
let rds_of_benchmark b =
let f = fun%workflow dest ->
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment