diff --git a/conv_net_classes.py b/conv_net_classes.py index a2af316..189f3f9 100644 --- a/conv_net_classes.py +++ b/conv_net_classes.py @@ -328,7 +328,7 @@ def errors(self, y): # check if y has same dimension of y_pred if y.ndim != self.y_pred.ndim: raise TypeError('y should have the same shape as self.y_pred', - ('y', target.type, 'y_pred', self.y_pred.type)) + ('y', y.type, 'y_pred', self.y_pred.type)) # check if y is of the correct datatype if y.dtype.startswith('int'): # the T.neq operator returns a vector of 0s and 1s, where 1