We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5724c70 commit 8c6b3afCopy full SHA for 8c6b3af
pyreason/scripts/learning/classification/classifier.py
@@ -49,7 +49,7 @@ def forward(self, x, t1: int = 0, t2: int = 0):
49
50
# Convert logits to probabilities assuming a multi-class classification.
51
probabilities = F.softmax(output, dim=1).squeeze()
52
- opts = self.integration_options
+ opts = self.interface_modes
53
54
# Prepare threshold tensor.
55
threshold = torch.tensor(opts.threshold, dtype=probabilities.dtype, device=probabilities.device)
0 commit comments