private-pgm icon indicating copy to clipboard operation
private-pgm copied to clipboard

Output model weighted mrgs

Open ceciliaferrando opened this issue 1 year ago • 1 comments

Adding a few changes to output AIM model in addition to synthetic data, and support weighted marginals in workload.

run() outputs the AIM model and the synthetic data, with option to specify number of synthetic data samples.

workload comprises (clique, weight) tuples.

compile_workload() accounts for weights in score computation.

In run(), user can provide custom initial_cliques (e.g. marginals containing target) for intelligent initialization. If None, 1-way marginals are used.

Added max_iters argument.

ceciliaferrando avatar Apr 01 '24 20:04 ceciliaferrando

Thanks for the PR, looks great!

ryan112358 avatar Apr 02 '24 02:04 ryan112358