def plot_precision_recall()

in mapillary_vistas/helpers/plot.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

def plot_precision_recall(precision, recall, average_precision, threshold, label, filename):
    fig, ax = plt.subplots()
    ax.plot(recall, precision)
    ax.set_xlabel('Recall')
    ax.set_ylabel('Precision')
    ax.set_ylim([0.0, 1.0])
    ax.set_xlim([0.0, 1.0])
    ax.set_title('Label: {}, Threshold: {:.2%}, AP: {:.2%}'.format(label, threshold, average_precision))
    fig.savefig(filename)
    plt.close(fig)