Skip to content

Commit bc1dd7a

Browse files
committed
Removed prints from entropy call func
1 parent 603c9be commit bc1dd7a

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

CollaborativeCoding/metrics/EntropyPred.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,14 +37,12 @@ def __call__(self, y_true: th.Tensor, y_logits: th.Tensor):
3737
f"y_logit class length: {y_logits.size(-1)}, expected: {self.num_classes}"
3838
)
3939
y_pred = nn.Softmax(dim=1)(y_logits)
40-
print(f"y_pred: {y_pred}")
4140
entropy_values = entropy(y_pred, axis=1)
4241
entropy_values = th.from_numpy(entropy_values)
4342

4443
# Fix numerical errors for perfect guesses
4544
entropy_values[entropy_values == th.inf] = 0
4645
entropy_values = th.nan_to_num(entropy_values)
47-
print(f"Entropy Values: {entropy_values}")
4846
for sample in entropy_values:
4947
self.stored_entropy_values.append(sample.item())
5048

0 commit comments

Comments
 (0)