Fix use report_confusion_matrix function

This commit is contained in:
allegroai 2023-12-03 11:10:57 +02:00
parent b3edd27e8d
commit 929710620a

View File

@ -46,7 +46,7 @@ def report_plots(logger, iteration=0):
# report confusion matrix # report confusion matrix
confusion = np.random.randint(10, size=(10, 10)) confusion = np.random.randint(10, size=(10, 10))
logger.report_matrix( logger.report_confusion_matrix(
"example_confusion", "example_confusion",
"ignored", "ignored",
iteration=iteration, iteration=iteration,
@ -56,7 +56,7 @@ def report_plots(logger, iteration=0):
) )
# report confusion matrix with 0,0 is at the top left # report confusion matrix with 0,0 is at the top left
logger.report_matrix( logger.report_confusion_matrix(
"example_confusion_0_0_at_top", "example_confusion_0_0_at_top",
"ignored", "ignored",
iteration=iteration, iteration=iteration,