diff --git a/code/expt/t-drive.py b/code/expt/t-drive.py index cdbfca8..a211fbb 100644 --- a/code/expt/t-drive.py +++ b/code/expt/t-drive.py @@ -93,7 +93,7 @@ def main(args): mae_u[i] += lmdk_bgt.mae(seq, rls_data_u)/args.iter # Adaptive - rls_data_a, _, _ = lmdk_bgt.adaptive(seq, lmdks, bgt['epsilon'], .5, .5) + rls_data_a, _, _ = lmdk_bgt.adaptive(seq, lmdks, bgt['epsilon'], .75, .25) mae_a[i] += lmdk_bgt.mae(seq, rls_data_a)/args.iter # Event @@ -150,12 +150,10 @@ def main(args): # Show plot # plt.show() # Save plot - lmdk_lib.save_plot(path + '.pdf') + lmdk_lib.save_plot(path + '-.pdf') print('[OK]', flush=True) - - def parse_args(): ''' Parse arguments.