слишком много индексов массив при рисовании кривой Roc для модели керас - PullRequest
Новая
       5

слишком много индексов массив при рисовании кривой Roc для модели керас

0 голосов
/ 11 февраля 2019

Я хочу классифицировать изображения, это около 7 классификаторов, и я хочу построить кривую ROC для измерения производительности модели

, она не сбалансирована, поэтому я пытаюсь сбалансировать набор поездов с помощью использованного увеличения данных ия хочу измерить производительность несколькими инструментами 

enter code here


fpr = dict()
tpr = dict()
roc_auc = dict()
for i in range(num_classes):
  fpr[i], tpr[i], _ = roc_curve(y_test[:, i], pred_test[:,i])
  roc_auc[i] = auc(fpr[i], tpr[i])

# Plot of a ROC curve for a specific class
for i in range(num_classes):
  plt.figure()
  plt.plot(fpr[i], tpr[i], label='ROC curve (area = %0.2f)' % roc_auc[i])
  plt.plot([0, 1], [0, 1], 'k--')
  plt.xlim([0.0, 1.0])
  plt.ylim([0.0, 1.05])
  plt.xlabel('False Positive Rate')
  plt.ylabel('True Positive Rate')
  plt.title('Receiver operating characteristic example')
  plt.legend(loc="lower right")
  plt.show()


 ---------------------------------------------------------------------------
 IndexError                                Traceback (most recent call last)
 <ipython-input-51-b620aaf312da> in <module>
  8 roc_auc = dict()
  9 for i in range(num_classes):
---> 10     fpr[i], tpr[i], _ = roc_curve(y_test[:, i], pred_test[:,i])
 11     roc_auc[i] = auc(fpr[i], tpr[i])
 12 

 IndexError: too many indices for array

ошибка показывает слишком много индексов y_test.shape дает мне (1103, 7) pred_test.shape дает мне (1103,) как я могу решить эту проблемунарисовать кривую Roc для каждого класса

...