From f38100a74f0d88cafeac8fb4c8fb99241007da08 Mon Sep 17 00:00:00 2001 From: pepperpepperpepper Date: Sun, 6 Dec 2015 23:16:09 -0800 Subject: started integrating probabilities --- ricky/params.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) (limited to 'ricky/params.py') 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): -- cgit v1.2.3-70-g09d2