happyz
/
propagation_gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update localization_log_gamma_model.py

This commit is contained in:
HappyZ 2019-06-07 14:07:28 -05:00
parent 40612ad801
commit 9be5c8630a
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
localization_log_gamma_model.py
View File

@ -55,6 +55,7 @@ def fittingSingle(filepath, args):
best_env_gamma = None best_env_gamma = None
best_rsses = None best_rsses = None
for i in range(args.iternum): for i in range(args.iternum):
try:
result = modelfit_log_gamma( result = modelfit_log_gamma(
rx_locs, rx_locs,
rx_rsses, rx_rsses,
@ -65,6 +66,8 @@ def fittingSingle(filepath, args):
monte_carlo_sampling=args.montecarlo, monte_carlo_sampling=args.montecarlo,
monte_carlo_sampling_rate=args.montecarlo_rate monte_carlo_sampling_rate=args.montecarlo_rate
) )
except RuntimeError:
continue
# unpack # unpack
fit_mse, est_tx_loc, est_tx_pwr, est_env_gamma, est_rsses = result fit_mse, est_tx_loc, est_tx_pwr, est_env_gamma, est_rsses = result
if fit_mse < min_fit_mse: if fit_mse < min_fit_mse:
@ -75,6 +78,9 @@ def fittingSingle(filepath, args):
best_env_gamma = est_env_gamma best_env_gamma = est_env_gamma
best_rsses = est_rsses best_rsses = est_rsses
if best_tx_loc is None:
return
# calculate diff # calculate diff
data_best_est = convert_vector_to_mat(rx_locs, best_rsses, data.shape) data_best_est = convert_vector_to_mat(rx_locs, best_rsses, data.shape)
data_diff = data - data_best_est data_diff = data - data_best_est