Skip to content

Commit

Permalink
Apply formatting
Browse files Browse the repository at this point in the history
Signed-off-by: Beat Buesser <[email protected]>
  • Loading branch information
beat-buesser committed Dec 16, 2024
1 parent 63b89ef commit 0b4bb68
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def predict(self, x: np.ndarray, batch_size: int = 128, verbose: bool = False, *
# get class counts
counts_pred = self._prediction_counts(x_i, batch_size=batch_size)
top = counts_pred.argsort()[::-1]
# conversion to int
# conversion to int
count1 = int(np.max(counts_pred))
count2 = int(counts_pred[top[1]])

Expand Down

0 comments on commit 0b4bb68

Please sign in to comment.