Skip to content
GitLab
Menu
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
f78346e3
Commit
f78346e3
authored
Dec 04, 2018
by
Lukas Platz
Browse files
KL_Energy: add keyword argument to mirror samples
parent
eaf710b7
Changes
1
Hide whitespace changes
Inline
Side-by-side
nifty5/minimization/kl_energy.py
View file @
f78346e3
...
...
@@ -8,7 +8,8 @@ from .. import utilities
class
KL_Energy
(
Energy
):
def
__init__
(
self
,
position
,
h
,
nsamp
,
constants
=
[],
constants_samples
=
None
,
_samples
=
None
):
constants_samples
=
None
,
mirror_samples
=
False
,
_samples
=
None
):
super
(
KL_Energy
,
self
).
__init__
(
position
)
if
h
.
domain
is
not
position
.
domain
:
raise
TypeError
...
...
@@ -17,11 +18,14 @@ 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
:
_samples
+=
tuple
(
-
s
for
s
in
_samples
)
self
.
_samples
=
_samples
self
.
_lin
=
Linearization
.
make_partial_var
(
position
,
constants
)
...
...
@@ -40,7 +44,8 @@ class KL_Energy(Energy):
def
at
(
self
,
position
):
return
KL_Energy
(
position
,
self
.
_h
,
0
,
self
.
_constants
,
self
.
_constants_samples
,
self
.
_samples
)
self
.
_constants_samples
,
self
.
_mirror_samples
,
self
.
_samples
)
@
property
def
value
(
self
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a 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