Commit 551b7bf0 authored by Philipp Arras's avatar Philipp Arras
Browse files

Fix plotting

parent 776602fb
......@@ -104,6 +104,8 @@ if __name__ == '__main__':
ESS_mean = []
R_hat_mean = []
if master:
fig, axes = plt.subplots(2, 2, num='results', figsize=(12, 8))
for i in range(50):
HMC.sample(10)
......@@ -119,8 +121,8 @@ if __name__ == '__main__':
if not master:
continue
plt.clf()
fig, axes = plt.subplots(2, 2, num='results', figsize=(12, 8))
for axx in axes.ravel():
axx.clear()
axes[0, 0].plot(data.val, 'kx', label='data')
axes[0, 0].plot(signal(mock_position).val, 'r-', label='ground truth')
axes[0, 0].plot(mean_result.val, 'k-', label='sample mean')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment