[BUGFIX] examples/cbc_iris.py works again

2021-06-15 15:59:47 +02:00
parent 1b420c1f6b
commit a37095409b
6 changed files with 39 additions and 125 deletions
--- a/prototorch/models/abstract.py
+++ b/prototorch/models/abstract.py
@@ -128,7 +128,7 @@ class SupervisedPrototypeModel(PrototypeModel):

    @property
    def num_classes(self):
-        return len(self.proto_layer.distribution)
+        return self.proto_layer.num_classes

    def compute_distances(self, x):
        protos, _ = self.proto_layer()