Skip to content

GitLab

Commit 39a7321a authored by Reimar H Leike's avatar Reimar H Leike
Browse files

changed number of samples and maxgradientnorm tolerance such that the pictures look nice again

parent 2add9edd
Pipeline #30895 passed with stages
in 1 minute and 28 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 4 deletions
demos/paper_demos/cartesian_wiener_filter.py
View file @ 39a7321a
......@@ -74,9 +74,11 @@ if __name__ == "__main__":
# Wiener filter
j = R.adjoint_times(N.inverse_times(data))
ctrl = ift.GradientNormController(name="inverter", tol_abs_gradnorm=0.1)
sampling_ctrl = ift.GradientNormController(name="sampling",tol_abs_gradnorm=1e2)
inverter = ift.ConjugateGradient(controller=ctrl)
sampling_inverter = ift.ConjugateGradient(controller=sampling_ctrl)
wiener_curvature = ift.library.WienerFilterCurvature(
S=S, N=N, R=R, inverter=inverter, sampling_inverter=inverter)
S=S, N=N, R=R, inverter=inverter, sampling_inverter=sampling_inverter)
m_k = wiener_curvature.inverse_times(j)
m = ht(m_k)
......@@ -88,7 +90,7 @@ if __name__ == "__main__":
ift.plot(data.cast_domain(plot_space), name='data.png', **plotdict)
ift.plot(m.cast_domain(plot_space), name='map.png', **plotdict)
# sampling the uncertainty map
mean, variance = ift.probe_with_posterior_samples(wiener_curvature, ht, 10)
mean, variance = ift.probe_with_posterior_samples(wiener_curvature, ht, 50)
ift.plot(ift.sqrt(variance).cast_domain(plot_space),
name="uncertainty.png", **plotdict)
ift.plot((mean+m).cast_domain(plot_space),
......
demos/paper_demos/wiener_filter.py
View file @ 39a7321a
......@@ -48,9 +48,11 @@ if __name__ == "__main__":
# Wiener filter
j = R.adjoint_times(N.inverse_times(data))
ctrl = ift.GradientNormController(name="inverter", tol_abs_gradnorm=1e-2)
sampling_ctrl = ift.GradientNormController(name="sampling",tol_abs_gradnorm=2e1)
inverter = ift.ConjugateGradient(controller=ctrl)
sampling_inverter = ift.ConjugateGradient(controller=sampling_ctrl)
wiener_curvature = ift.library.WienerFilterCurvature(
S=S, N=N, R=R, inverter=inverter, sampling_inverter=inverter)
S=S, N=N, R=R, inverter=inverter, sampling_inverter=sampling_inverter)
m_k = wiener_curvature.inverse_times(j)
m = ht(m_k)
......@@ -60,6 +62,6 @@ if __name__ == "__main__":
ift.plot(m, name="map.png", **plotdict)
# sampling the uncertainty map
mean, variance = ift.probe_with_posterior_samples(wiener_curvature, ht, 5)
mean, variance = ift.probe_with_posterior_samples(wiener_curvature, ht, 50)
ift.plot(ift.sqrt(variance), name="uncertainty.png", **plotdict)
ift.plot(mean+m, name="posterior_mean.png", **plotdict)
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