Skip to content

Commit

Permalink
Merge branch 'dev_1.19.0' into sklearn_nbclasses
Browse files Browse the repository at this point in the history
  • Loading branch information
beat-buesser authored Dec 16, 2024
2 parents 617d8d9 + 0b4bb68 commit bd58b1a
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 bd58b1a

Please sign in to comment.