Skip to content

Commit 8c6b3af

Browse files
committed
bug in classifier.py
1 parent 5724c70 commit 8c6b3af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyreason/scripts/learning/classification/classifier.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def forward(self, x, t1: int = 0, t2: int = 0):
4949

5050
# Convert logits to probabilities assuming a multi-class classification.
5151
probabilities = F.softmax(output, dim=1).squeeze()
52-
opts = self.integration_options
52+
opts = self.interface_modes
5353

5454
# Prepare threshold tensor.
5555
threshold = torch.tensor(opts.threshold, dtype=probabilities.dtype, device=probabilities.device)

0 commit comments

Comments
 (0)