def evaluate_with_ratio(test_ratio):
flags = flag_reader.read_flag()
evaluatemain(flags,
eval_forward=False,
test_ratio=test_ratio,
plot_histo=False)
plotsAnalysis.SpectrumComparisonNGeometryComparison(
3, 2, (13, 8), flags.model_name, flags.boundary)
stop_thres=flags.stop_threshold,
verb_step=flags.verb_step,
model_name=flags.model_name,
write_summary=True,
eval_forward=eval_forward,
time_recorder=TK)
else:
pred_file = save_file
truth_file = os.path.join(os.path.abspath(''), 'data',
'test_truth.csv')
mae, mse = compare_truth_pred(pred_file, truth_file)
plt.figure(figsize=(12, 6))
plt.hist(mse, bins=100)
plt.xlabel('Mean Squared Error')
plt.ylabel('cnt')
plt.suptitle('Backprop (Avg MSE={:.4e})'.format(np.mean(mse)))
plt.savefig(
os.path.join(os.path.abspath(''), 'data',
'Backprop_{}.png'.format(flags.model_name)))
plt.show()
print('Backprop (Avg MSE={:.4e})'.format(np.mean(mse)))
if __name__ == '__main__':
flags = flag_reader.read_flag()
evaluatemain(flags, eval_forward=False)
plotsAnalysis.SpectrumComparisonNGeometryComparison(
3, 2, (13, 8), flags.model_name)