Skip to content
Snippets Groups Projects
Commit ef60a287 authored by Lukas Platz's avatar Lukas Platz
Browse files

changed argument to 'gen_mirrored_samples' for clarity

not storing the value because it is not used after the
first initialization.
parent f78346e3
No related branches found
No related tags found
No related merge requests found
......@@ -8,7 +8,7 @@ from .. import utilities
class KL_Energy(Energy):
def __init__(self, position, h, nsamp, constants=[],
constants_samples=None, mirror_samples=False,
constants_samples=None, gen_mirrored_samples=False,
_samples=None):
super(KL_Energy, self).__init__(position)
if h.domain is not position.domain:
......@@ -18,13 +18,12 @@ class KL_Energy(Energy):
if constants_samples is None:
constants_samples = constants
self._constants_samples = constants_samples
self._mirror_samples = mirror_samples
if _samples is None:
met = h(Linearization.make_partial_var(
position, constants_samples, True)).metric
_samples = tuple(met.draw_sample(from_inverse=True)
for _ in range(nsamp))
if mirror_samples:
if gen_mirrored_samples:
_samples += tuple(-s for s in _samples)
self._samples = _samples
......@@ -43,9 +42,9 @@ class KL_Energy(Energy):
self._metric = None
def at(self, position):
return KL_Energy(position, self._h, 0, self._constants,
self._constants_samples, self._mirror_samples,
self._samples)
return KL_Energy(position, self._h, 0,
self._constants, self._constants_samples,
_samples = self._samples)
@property
def value(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment