summaryrefslogtreecommitdiff
path: root/ricky/params.py
diff options
context:
space:
mode:
Diffstat (limited to 'ricky/params.py')
-rw-r--r--ricky/params.py18
1 files changed, 15 insertions, 3 deletions
diff --git a/ricky/params.py b/ricky/params.py
index c5595d4..4f91776 100644
--- a/ricky/params.py
+++ b/ricky/params.py
@@ -18,10 +18,22 @@ class Params(object):
"""string representation"""
return pprint.pformat(self.as_dict())
- def randomize(self):
- """assign random values to all params, taking into account weight"""
+ def randomize(
+ self,
+ probabilities=None,
+ probabilities_local=False
+ ):
+ """assign random values to all params
+ if using a probabilities.json file, weight is taken
+ into account"""
+ if probabilities:
+ probabilities = self._load_probabilities(probabilities)
+ else if probabilities_local:
+ probabilities = self._load_probabilities(probabilities_local)
+ else:
+ probabilities = {}
for param in self._params:
- param.randomize()
+ param.randomize(probability=probabilities.get(param.name))
@property
def api(self):