HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix print() for ConfusionMatrix for Classify task (#19169)

Browse source 
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
This commit is contained in:
Mohammed Yasin 2025-02-13 12:39:26 +08:00 • committed by GitHub
parent d6a07ee78a
commit 710b40dcb1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/utils/metrics.py
View file

@ -440,7 +440,7 @@ class ConfusionMatrix:
def print(self): def print(self):
"""Print the confusion matrix to the console.""" """Print the confusion matrix to the console."""
for i in range(self.nc + 1): for i in range(self.matrix.shape[0]):
LOGGER.info(" ".join(map(str, self.matrix[i]))) LOGGER.info(" ".join(map(str, self.matrix[i])))