【Python】绘制PR曲线

数据挖掘模型训练好了之后，对于二分类问题，可以预测每个测试样本属于某个类别的概率，
当我们设置不同阈值时，将得到不同的P值（Precision，查准率）和R值（Recall，查全率）
基于此，可以绘制PR曲线，python绘制PR曲线比较方便。
首先，可以使用precision_recall_curve函数得到

#选择测试数据
import pandas as pddf = pd.read_csv('file_name.csv')
X = df[df.columns[:-1]]
y = df[df.columns[-1]]#训练——测试数据集划分
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X[y < 2], y[y < 2],test_size=.5,random_state=1)#构造模型
from sklearn.ensemble import RandomForestClassifier
model = RandomForestClassifier()
model.fit(X_train, y_train)
#model.predict_proba将返回属于各个类别的概率，每行概率和为1
y_prob = model.predict_proba(X_test)[:,1]#计算P值和R值：precision_recall_curve函数返回值分别为查准率precision，查全率recall，以及对应的阈值thresholdsprecision, recall, thresholds = sklearn.metrics.precision_recall_curve(y_test, y_prob, pos_label=None, sample_weight=None)#然后根据precision, recall绘制PR曲线plt.plot()

其次可以直接使用plot_precision_recall_curve函数绘制PR曲线

from sklearn.metrics import plot_precision_recall_curve
disp = plot_precision_recall_curve(classifier, X_test, y_test)
disp.ax_.set_title('2-class Precision-Recall curve')

【Python】绘制PR曲线相关推荐

机器学习：python绘制P-R曲线与ROC曲线
Python绘制P-R曲线与ROC曲线查准率与查全率 P-R曲线的绘制 ROC曲线的绘制查准率与查全率 P-R曲线,就是查准率(precision)与查全率(recall)的曲线,以查准率作为 ...
python画pr曲线代码_Yolov3测试图及绘制PR曲线,yoloV3,map,和,画
训练指令:./darknet.exe detector train data/obj.data yolo-obj.cfg darknet53.conv.74 | tee train_yolov3.lo ...
yolo模型(四)绘制PR曲线所用到的reval_voc_py.py代码
在绘制PR曲线时要自己生成一个pkl文件,需要reval_voc_py.py代码来生.现贴出基于python2和python3的代码,按照自己的python版本直接复制即可.拿走不谢~ reval_v ...
python画曲线-python绘制动态曲线教程
从txt种获取数据并且通过动态曲线显示 import numpy as np import matplotlib.pyplot as plt import matplotlib.animation ...
Python绘制简单函数曲线（包括坐标范围限制、刻度指定）
python绘制简单函数曲线,包括坐标范围限制.刻度指定 ################### 二维曲线图 ################## import numpy as np import ...
机器学习入门案例：鸢尾花数据集分类绘制PR曲线
案例使用鸢尾花数据集进行分类预测,并绘制评价分类性能的PR曲线图认识分类任务和数据集 Iris(鸢尾花)数据集案例演示中使用的是有监督的机器学习算法:SVM 支持向量机建立模型的流程如下: 训练 ...
基于python绘制ROC曲线
基于python绘制ROC曲线,直接附代码: from sklearn.metrics import roc_curve, auc from sklearn.model_selection impor ...
Python绘制loss曲线、准确率曲线
Python 绘制 loss 曲线.准确率曲线使用 python 绘制网络训练过程中的的 loss 曲线以及准确率变化曲线,这里的主要思想就时先把想要的损失值以及准确率值保存下来,保存到 .txt ...
python绘制pr曲线图_利用Python中的numpy包实现PR曲线和ROC曲线的计算!
闲来无事,边理解PR曲线和ROC曲线,边写了一下计算两个指标的代码.在python环境下,sklearn里有现成的函数计算ROC曲线坐标点,这里为了深入理解这两个指标,写代码的时候只用到numpy包. ...
python画pr曲线_python 画函数曲线示例
python 画函数曲线示例如下所示: import numpy as np import matplotlib.pyplot as plt x = np.linspace(0, 2 * np.pi ...

【Python】绘制PR曲线

【Python】绘制PR曲线相关推荐

最新文章

热门文章