Patent attributes
A method, system, and computer program product for visualizing a machine learning model are provided. A confusion matrix and model performance metric data are received from a classification model. For each data point in the confusion matrix, a corresponding pixel is generated. The pixels are grouped into clusters. Each cluster represents a label in the confusion matrix. A centroid is generated for each cluster. Using the model performance metric data, a misclassification indicator arrow is generated for each misclassified data point. The misclassification indicator arrow indicates both the predicted class and the actual class. The clusters, the centroids, and the misclassification indicator arrow are displayed as a graphical visualization of the machine learning model.
