anton/mlpractical
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add evaluation phase for stochastic layers

Browse Source
This commit is contained in:
AntreasAntoniou 2017-11-07 12:55:23 +00:00
parent 1021afa640
commit a1d92fb92e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlp/optimisers.py
View File

@ -80,7 +80,7 @@ class Optimiser(object):
data_mon_vals = OrderedDict([(key + label, 0.) for key data_mon_vals = OrderedDict([(key + label, 0.) for key
in self.data_monitors.keys()]) in self.data_monitors.keys()])
for inputs_batch, targets_batch in dataset: for inputs_batch, targets_batch in dataset:
activations = self.model.fprop(inputs_batch) activations = self.model.fprop(inputs_batch, evaluation=True)
for key, data_monitor in self.data_monitors.items(): for key, data_monitor in self.data_monitors.items():
data_mon_vals[key + label] += data_monitor( data_mon_vals[key + label] += data_monitor(
activations[-1], targets_batch) activations[-1], targets_batch)