diff options
Diffstat (limited to 'simulation/utils_blocks.py')
| -rw-r--r-- | simulation/utils_blocks.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/simulation/utils_blocks.py b/simulation/utils_blocks.py index 72a6881..00b429e 100644 --- a/simulation/utils_blocks.py +++ b/simulation/utils_blocks.py @@ -37,7 +37,10 @@ class ActiveLearning(be.SimpleExtension): self.params = params def do(self, which_callback, *args): - exp_out_par = np.exp(np.sum(self.params, axis=1)) + try: + exp_out_par = np.exp(np.sum(self.params.get_value(), axis=1)) + except AttributeError: + exp_out_par = np.exp(np.sum(self.params, axis=1)) self.dataset.node_p = exp_out_par / np.sum(exp_out_par) |