expt_lmdk_sel: Again

This commit is contained in:
Manos Katsomallos 2021-10-05 05:32:48 +02:00
parent 438148edf7
commit dafe129a97

View File

@ -61,9 +61,10 @@ def main(args):
# Turn part to histogram # Turn part to histogram
hist_pt, _ = lmdk_lib.get_hist(pt, lmdks_pt) hist_pt, _ = lmdk_lib.get_hist(pt, lmdks_pt)
# Get an option for this part # Get an option for this part
res_pt = opts[0]
if len(opts) > 1: if len(opts) > 1:
res_pt, _ = exp_mech.exponential(hist_pt, opts, exp_mech.score, 1.0, e) res_pt, _ = exp_mech.exponential(hist_pt, opts, exp_mech.score, 1.0, e)
elif len(opts) > 0:
res_pt = opts[0]
# Merge options of all parts # Merge options of all parts
res[pt_i] = np.sum(res_pt) res[pt_i] = np.sum(res_pt)
# Calculate MAE # Calculate MAE