matplotlib如何每5个epoch显示一个点

您可以使用matplotlib的FuncAnimation模块来实现每5个epoch显示一个点的需求。

下面是一个简单的示例代码，假设您的训练共有100个epoch，每个epoch的损失值保存在一个列表losses中：

import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

fig, ax = plt.subplots()
line, = ax.plot([], [], 'bo')
ax.set_xlim([0, 100])
ax.set_ylim([0, max(losses)])

def update(i):
    if (i+1)%5 == 0:
        x = range(1, i+2, 5)
        y = losses[::5][:len(x)]
        line.set_data(x, y)
    return line,

ani = FuncAnimation(fig, update, frames=100, blit=True)
plt.show()

在这个示例中，我们定义了一个update函数，该函数会在每个动画帧中被调用。在每5个epoch时，我们会更新x轴和y轴的数据，并将它们赋值给line对象的数据。最后，我们将line对象返回以更新图形。

调用FuncAnimation函数时，我们指定了动画的总帧数为100，即总epoch数。blit参数指定是否使用快速绘制技术，加速动画的绘制。最后，我们调用plt.show()函数以显示图形。

请注意，由于动画会在每个epoch中更新，因此它将会比训练过程本身慢得多。如果您的训练非常缓慢，动画可能会导致整个过程变得更慢。在这种情况下，您可以考虑将动画的帧率降低，或者在训练过程中记录更少的epoch。