def plot_to_pdf(Y_valid, y_predict_probs, filename):
"""
Plots the predict proba and precision recall
curve on a single graph
"""
with PdfPages(filename + '.pdf') as pdf:
y_predict = y_predict_probs
inclf = filename
plot_predict_proba(y_predict, inclf, pdf=pdf)
plot_precision_recall_n(Y_valid.values.ravel(),
y_predict_probs, inclf, pdf=pdf)
评论列表
文章目录