def setUp(self):
self._metric_function = Mock(return_value='test')
self._metric = BatchLambda('test', self._metric_function)
self._states = [{
torchbearer.Y_TRUE: Variable(torch.FloatTensor([1])),
torchbearer.Y_PRED: Variable(torch.FloatTensor([2]))
}, {
torchbearer.Y_TRUE: Variable(torch.FloatTensor([3])),
torchbearer.Y_PRED: Variable(torch.FloatTensor([4]))
}, {
torchbearer.Y_TRUE: Variable(torch.FloatTensor([5])),
torchbearer.Y_PRED: Variable(torch.FloatTensor([6]))
}]
def build(self):
if on_epoch:
return EpochLambda(name, metric_function)
else:
return BatchLambda(name, metric_function)
def decorator(metric_function):
if on_epoch:
return EpochLambda(name, metric_function)
else:
return BatchLambda(name, metric_function)