plot_accuracy.py 文件源码

def plot_accuracy_curve(title, style, y, vp_size, smooth, interval, maxlen, **kwargs):
    mv_avg = kwargs.pop('mv_avg', False)

    x = range(vp_size, 1 + len(y) * vp_size, vp_size)

    min_len = min(len(x), len(y), maxlen)
    x, y = x[:min_len], y[:min_len]

    if mv_avg is not False:
        y = move_avg(y, mv_avg)

    if smooth > 0:
        try:
            x_new = np.linspace(min(x), max(x), smooth)
        except ValueError as e:
            print('Plot error:', title, e)
            return
        power_smooth = spline(x, y, x_new)
    else:
        x_new = x
        power_smooth = y

    x_new = pick_interval(x_new, interval)
    power_smooth = pick_interval(power_smooth, interval)

    plt.plot(x_new, power_smooth, style, label=title, **kwargs)