公开(公告)号：US20240273386A1

公开(公告)日：2024-08-15

申请号：US18436182

申请日：2024-02-08

Applicant: McMaster University

Inventor： Mohammadreza Heydarian Firouz Aba ,  Thomas Edward Doyle ,  Reza Samavi

IPC: G06N5/04

CPC classification number: G06N5/04

Abstract: A system, method and computer program product for evaluating a multi-label classification machine learning model. A model labelled dataset is received from the model and includes a plurality of data elements each labelled with zero or more model predicted labels of q potential classes. A multi-label confusion matrix is defined to include q+1 rows with q rows for true labels and 1 row for no true label and q+1 columns with q columns for predicted labels and 1 column for no predicted label. The matrix is populated by comparing the model labelled dataset with a true labelled dataset. At least one performance metric is calculated from the populated multi-label confusion matrix.