使用Matplotlib库中的函数绘制精确率-召回率曲线。
示例代码:
import matplotlib.pyplot as plt
from sklearn.metrics import precision_recall_curve
def plot_precision_recall_curve(y_true, y_score):
precision, recall, thresholds = precision_recall_curve(y_true, y_score)
# 计算精确率、召回率和阈值
plt.plot(recall, precision)
# 绘制精确率-召回率曲线
plt.xlabel('Recall')
plt.ylabel('Precision')
plt.title('Precision-Recall Curve')
plt.show()
使用示例:
# 假设y_true和y_score是模型预测结果的真实标签和预测分数
plot_precision_recall_curve(y_true, y_score)
此时,将会显示出一个精确率-召回率曲线图。