Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
ift
NIFTy
Commits
c40da307
Commit
c40da307
authored
Jul 15, 2021
by
Philipp Arras
Browse files
KL energies: actually use napprox
parent
ea74ecaa
Changes
1
Show whitespace changes
Inline
Side-by-side
src/minimization/kl_energies.py
View file @
c40da307
...
@@ -407,7 +407,7 @@ def MetricGaussianKL(mean, hamiltonian, n_samples, mirror_samples, constants=[],
...
@@ -407,7 +407,7 @@ def MetricGaussianKL(mean, hamiltonian, n_samples, mirror_samples, constants=[],
_
,
ham_sampling
=
_reduce_by_keys
(
mean
,
hamiltonian
,
point_estimates
)
_
,
ham_sampling
=
_reduce_by_keys
(
mean
,
hamiltonian
,
point_estimates
)
sampler
=
_MetricGaussianSampler
(
mean
,
ham_sampling
,
n_samples
,
sampler
=
_MetricGaussianSampler
(
mean
,
ham_sampling
,
n_samples
,
mirror_samples
)
mirror_samples
,
napprox
)
local_samples
=
sampler
.
draw_samples
(
comm
)
local_samples
=
sampler
.
draw_samples
(
comm
)
mean
,
hamiltonian
=
_reduce_by_keys
(
mean
,
hamiltonian
,
constants
)
mean
,
hamiltonian
=
_reduce_by_keys
(
mean
,
hamiltonian
,
constants
)
...
@@ -517,7 +517,8 @@ def GeoMetricKL(mean, hamiltonian, n_samples, minimizer_samp, mirror_samples,
...
@@ -517,7 +517,8 @@ def GeoMetricKL(mean, hamiltonian, n_samples, minimizer_samp, mirror_samples,
_
,
ham_sampling
=
_reduce_by_keys
(
mean
,
hamiltonian
,
point_estimates
)
_
,
ham_sampling
=
_reduce_by_keys
(
mean
,
hamiltonian
,
point_estimates
)
sampler
=
_GeoMetricSampler
(
mean
,
ham_sampling
,
minimizer_samp
,
sampler
=
_GeoMetricSampler
(
mean
,
ham_sampling
,
minimizer_samp
,
start_from_lin
,
n_samples
,
mirror_samples
)
start_from_lin
,
n_samples
,
mirror_samples
,
napprox
)
local_samples
=
sampler
.
draw_samples
(
comm
)
local_samples
=
sampler
.
draw_samples
(
comm
)
mean
,
hamiltonian
=
_reduce_by_keys
(
mean
,
hamiltonian
,
constants
)
mean
,
hamiltonian
=
_reduce_by_keys
(
mean
,
hamiltonian
,
constants
)
return
_SampledKLEnergy
(
mean
,
hamiltonian
,
sampler
.
n_eff_samples
,
False
,
return
_SampledKLEnergy
(
mean
,
hamiltonian
,
sampler
.
n_eff_samples
,
False
,
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment