aboutsummaryrefslogtreecommitdiffstats
path: root/utils/misc.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils/misc.py')
-rw-r--r--utils/misc.py18
1 files changed, 17 insertions, 1 deletions
diff --git a/utils/misc.py b/utils/misc.py
index 735f95b..6a81bbf 100644
--- a/utils/misc.py
+++ b/utils/misc.py
@@ -25,13 +25,15 @@ from utils.enums import Likelihood, ParamTag
class Param(object):
"""Parameter class to store parameters.
"""
- def __init__(self, name, value, ranges, std=None, tex=None, tag=None):
+ def __init__(self, name, value, ranges, seed=None, std=None, tex=None, tag=None):
+ self._seed = None
self._ranges = None
self._tex = None
self._tag = None
self.name = name
self.value = value
+ self.seed = seed
self.ranges = ranges
self.std = std
self.tex = tex
@@ -46,6 +48,16 @@ class Param(object):
self._ranges = [val for val in values]
@property
+ def seed(self):
+ if self._seed is None: return self.ranges
+ return tuple(self._seed)
+
+ @seed.setter
+ def seed(self, values):
+ if values is None: return
+ self._seed = [val for val in values]
+
+ @property
def tex(self):
return r'{0}'.format(self._tex)
@@ -137,6 +149,10 @@ class ParamSet(Sequence):
return tuple([obj.value for obj in self._params])
@property
+ def seeds(self):
+ return tuple([obj.seed for obj in self._params])
+
+ @property
def ranges(self):
return tuple([obj.ranges for obj in self._params])