utils.py 文件源码

def plot_acc_line(k, style, xs, ys, interval=1, smooth=300):
    dx, dy = xs[k], ys[k]

    if interval > 1:
        dx = [e for i, e in enumerate(dx) if (i + 1) % interval == 0]
        dy = [e for i, e in enumerate(dy) if (i + 1) % interval == 0]

    min_len = min(len(dx), len(dy))
    dx, dy = dx[:min_len], dy[:min_len]

    x_new = np.linspace(min(dx), max(dx), smooth)
    power_smooth = spline(dx, dy, x_new)

    plt.plot(x_new, power_smooth, linewidth=2.0, label=k, linestyle=style)