[机器学习笔记] Python 绘制ROC曲线的例子

Python 绘制ROC曲线的例子

ROC的全称是“受试者工作特征”（Receiver Operating Characteristic）曲线。

根据学习器的预测结果对样例进行排序，按此顺序逐个把样本作为正例进行预测，每次计算出两个重要量的值（TPR、FPR），分别以它们为横、纵坐标作图。与PR曲线使用查准率、查全率为纵、横不同，ROC 曲线的纵轴是“真正例率”（True Positive Rate,TTR），横轴是“假正例率”（False Positive Rate，FPR）。

如图所示：

下面使用Python来作出该图：

# -*- coding: utf-8 -*-
"""
Created on Tue Mar 24 19:04:21 2020

@author: Bean029
"""

import numpy as np
import matplotlib.pyplot as plt
from itertools import cycle

from sklearn import svm, datasets
from sklearn.metrics import roc_curve, auc
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import label_binarize
from sklearn.multiclass import OneVsRestClassifier
from scipy imp