diff --git a/localization_log_gamma_model.py b/localization_log_gamma_model.py index d5d0a6b..bf85e3a 100644 --- a/localization_log_gamma_model.py +++ b/localization_log_gamma_model.py @@ -40,7 +40,7 @@ def fittingSingle(filepath, args): best_tx_pwr = None best_env_gamma = None best_rsses = None - for i in range(1000): + for i in range(args.iternum): result = modelfit_log_gamma( rx_locs, rx_rsses, @@ -48,8 +48,8 @@ def fittingSingle(filepath, args): bounds_gamma=(2, 6), bounds_loc_x=(0, 6.4), bounds_loc_y=(0, 6.4), - monte_carlo_sampling=True, - monte_carlo_sampling_rate=0.8 + monte_carlo_sampling=args.montecarlo, + monte_carlo_sampling_rate=args.montecarlo_rate ) # unpack fit_mse, est_tx_loc, est_tx_pwr, est_env_gamma, est_rsses = result @@ -136,6 +136,27 @@ if __name__ == '__main__': default=False, help='enable visualize' ) + p.add_argument( + '--montecarlo', '-m', + dest='montecarlo', + action='store_true', + default=False, + help='enable monte carlo sampling' + ) + p.add_argument( + '--montecarlo-rate', '-mr', + dest='montecarlo_rate', + type=float, + default=0.8, + help='if monte carlo sampling enabled, specify the rate (0-1, default 0.8)' + ) + p.add_argument( + '--iternum', '-n', + dest='iternum', + type=int, + default=1000, + help='num of iterations' + ) p.add_argument( '--output', '-o', dest='outputfile',