CODEWITHSUNDEEP
cafferocaucpycaffematcaffe
khan
khan

Reputation: 531

Calculating ROC and AUC in Caffe?

I have trained imagenet in Caffe. Now i am trying to calculate ROC/AUC for my model and the trained model provided by caffe. I have two questions:

1) ROC/AUC is mainly used for binary classes, but i also found that in some cases people used it for multi-classes. Is it possible for 1000 classes. And what will be its impact? As in reviews people didn't give good answer for ROC/AUC in multi-class problems.

2) If possible, and comparing two models based on ROC/AUC will be a good idea, Can anybody tell how to do it for these 1000 classes in Caffe? And do i have to retrain the models from scratch, or can i calculate only with final trained models?

Regards

Upvotes: 3

Views: 777

Answers (1)

Dmytro Prylipko
Dmytro Prylipko

Reputation: 5084

This discussion addresses multi-class ROC/AUC analysis nicely. Answering your questions:

  1. You can do multiple one-vs-all classifications for each class, thus building multiple ROC curves.

  2. Having computed 1000 AUC values, you can come up with the mean AUC over all classes and use this metric to compare goodness of your models. No, you don't need to retrain your models.

Also, pay an attention that ROC/AUC metrics are quite specific and used mostly in detection/biometry tasks like voice identification.

Upvotes: 0

Related Questions