diff options
Diffstat (limited to 'ricky/params.py')
| -rw-r--r-- | ricky/params.py | 18 |
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): |
