如何用matplotlib 在一张图上画多条 precision recall curve PR曲线

搬运自stackoverflow:

sklearn 的 display 函数基本上都是 matplotlib 函数直接加上个外壳，所以，用ax就可以： 

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
PrecisionRecallDisplay.from_predictions(y_train, y_pred_train, ax=ax)
PrecisionRecallDisplay.from_predictions(y_test, y_pred, ax=ax)
plt.show()

另外如果像我一样，上述代码在 jupyter notebook 上无法显示图像的，在

import matplotlib.pyplot as plt

前加一句

%matplotlib inline
import matplotlib.pyplot as plt

然后直接用

fig

而不是

fig.show()
#或者是
plt.plot()

即可。