Fix __repr__ implementation for SingleLayerModel (#46)
This commit is contained in:
parent
d41063dca6
commit
6892424a43
@ -69,7 +69,7 @@ class SingleLayerModel(object):
|
||||
return self.layer.params_penalty()
|
||||
|
||||
def __repr__(self):
|
||||
return 'SingleLayerModel(' + str(layer) + ')'
|
||||
return 'SingleLayerModel(' + str(self.layer) + ')'
|
||||
|
||||
|
||||
class MultipleLayerModel(object):
|
||||
|
Loading…
Reference in New Issue
Block a user