Multiclass classification To better illustrate the differences, let us assume that your goal is that of classifying SO questions into n_classes different, mutually exclusive classes. For the sake of simplicity in this example we will only consider four classes, namely ‘Python’, ‘Java’, ‘C++’ and ‘Other language’. Let us assume that you have a dataset formed … Read more
Your input to confusion_matrix must be an array of int not one hot encodings. matrix = metrics.confusion_matrix(y_test.argmax(axis=1), y_pred.argmax(axis=1))
As people mentioned in comments you have to convert your problem into binary by using OneVsAll approach, so you’ll have n_class number of ROC curves. A simple example: from sklearn.metrics import roc_curve, auc from sklearn import datasets from sklearn.multiclass import OneVsRestClassifier from sklearn.svm import LinearSVC from sklearn.preprocessing import label_binarize from sklearn.model_selection import train_test_split import matplotlib.pyplot … Read more