python 画roc曲线

label.txt:

1
0
1
0
1
0

feature.txt

0.1
0.2
0.2
0.2
0.2
0.3

代码：

  

   

    

     

    
    

     

      # -*- coding: utf-8 -*-
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import sklearn
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import matplotlib.pyplot as plt
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import skimage
     
    
   

    

     

    
    

     

      import sklearn.metrics.pairwise as pw
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #读取标签文件
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def read_labels(labelfile):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          fin=open(labelfile)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          lines=fin.readlines()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          labels=np.empty((len(lines),))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          k=0;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    for line in lines:
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              labels[k]=int(line)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              k=k+1;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          fin.close()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    return labels
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def read_Feautures(labelfile):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          fin=open(labelfile)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          lines=fin.readlines()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          labels=np.empty((len(lines),))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          k=0;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    for line in lines:
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              labels[k]=float(line)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              k=k+1;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          fin.close()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    return labels
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #画ROC曲线图
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def draw_roc_curve(fpr1,tpr1,fpr2,tpr2, title='cosine',save_name='roc_lfw'):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.figure()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.plot(fpr1, tpr1,'r')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.plot(fpr2, tpr2,'g')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.plot([0, 1], [0, 1], 'k--')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.xlim([0.0, 1.0])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.ylim([0.0, 1.0])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.xlabel('FPR')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.ylabel('TPR')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.title('ROC(Receiver operating characteristic)using: '+title)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.legend(loc="lower right")
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.show()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          plt.savefig(save_name+'.png')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if __name__=='__main__':
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          labels = read_labels(u"label.txt")
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          predicts = read_Feautures(u"feature.txt")
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          fpr1, tpr1, threshold1s=sklearn.metrics.roc_curve(labels,predicts)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    print(threshold1s)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    #draw_roc_curve(fpr1,tpr1,title='JSS',save_name='lfw_evaluate')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          labels = read_labels(u"label.txt")
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          predicts = read_Feautures(u"feature.txt")
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          fpr2, tpr2, threshold2s=sklearn.metrics.roc_curve(labels,predicts)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    print(threshold2s)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          draw_roc_curve(fpr1,tpr1,fpr2,tpr2,title='AS',save_name='lfw_evaluate')
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/120818500