diff --git a/code/expt/expt_lmdk_sel.py b/code/expt/expt_lmdk_sel.py index a273d2a..7e84939 100644 --- a/code/expt/expt_lmdk_sel.py +++ b/code/expt/expt_lmdk_sel.py @@ -61,8 +61,8 @@ def main(args): # Turn part to histogram hist_pt, _ = lmdk_lib.get_hist(pt, lmdks_pt) # Get an option for this part - res_pt = np.array([]) - if len(opts) > 0: + res_pt = opts[0] + if len(opts) > 1: res_pt, _ = exp_mech.exponential(hist_pt, opts, exp_mech.score, 1.0, e) # Merge options of all parts res[pt_i] = np.sum(res_pt)