Tractables/Dice.jl

fix. allow us to register_stats! to rs, or create MLELossMgr

Closed this issue · 1 comments

github-actions commented

Also save distribution of metric being trained

function f_emit′(tag)

println_flush(rs.io, "Saving $(tag) distribution...")

time_infer = @Elapsed metric_dist = pr_mixed(rs.var_vals)(metric)

println(rs.io, " $(time_infer) seconds")

save_metric_dist(joinpath(rs.out_dir, "dist_$(name(p.metric))_$(tag).csv"), metric_dist; rs.io)

println(rs.io)

end

Dice.jl/examples/qc/benchmarks/benchmarks.jl

Line 475 in 1c0c443

# TODO: fix. allow us to register_stats! to rs, or create MLELossMgr 

    println(rs.io, "  $(time_build_loss) seconds")
    println(rs.io)

    SimpleLossMgr(loss)

    # TODO: fix. allow us to register_stats! to rs, or create MLELossMgr
    # # Also save distribution of metric being trained
    # function f_emit′(tag)
    #     println_flush(rs.io, "Saving $(tag) distribution...")
    #     time_infer = @elapsed metric_dist = pr_mixed(rs.var_vals)(metric)
    #     println(rs.io, "  $(time_infer) seconds")
    #     save_metric_dist(joinpath(rs.out_dir, "dist_$(name(p.metric))_$(tag).csv"), metric_dist; rs.io)
    #     println(rs.io)

    #     emit_stats(mgr, tag)
    # end
end

struct TreeSize <: Metric{BST} end