error = {}
rmsedata = {}
amedata = {}
for mem in mem_list:
rmseloc = {}
ameloc = {}
for loc in loc_list:
rmse = []
ame = []
for wgt in wgt_list:
filename = 'h3dw%2.2dl%2.2dm%2.2d/output_analy.txt' % (
wgt, loc, mem)
# analysis
analy = ev.get_data(filename)
# get errors
rms, am = ev.errors(truth, analy)
print('%s : RMS = %3.4f , AM = %3.4f' % (filename, rms, am))
rmse.append(rms)
ame.append(am)
rmseloc[str(loc)] = rmse
ameloc[str(loc)] = ame
rmsedata[str(mem)] = rmseloc
amedata[str(mem)] = ameloc
error['rmse'] = rmsedata
error['ame'] = amedata
with open(jsonfile, 'w') as f:
json.dump(error, f)
if plot_fig:
import matplotlib.pyplot as plt
