search

duggalrahul / AlexNet-Experiments-Keras

Code examples for training AlexNet using Keras and Theano
MIT License
107 stars 53 forks source link

Generation of confusion matrix and roc curve #11

Open ghost opened 6 years ago

ghost commented 6 years ago

Many thanks for your post. I’m following your alexnet code as a feature extractor for my binary medical image classification. I’m using tensorflow backend. Could you please let know the code to compute confusion matrix and roc auc scores? It would be extremely helpful.

ghost commented 6 years ago

Your reply with the code will be more than useful. Thanks once again.

duggalrahul commented 6 years ago

Hi @raaju-shiv , If you can obtain the output prediction of the model for your own data, it is not very difficult to draw the ROC curve. You can checkout this video to build an intuitive understanding of what these metrics are. The code will be pretty straightforward once you understand the concept. Let me know if you still have any difficulties.