def draw_loss(n_epochs, losses, val_losses): epochs_range = range(n_epochs) fig1=plt.figure(figsize=(4,3)) plt.plot(epochs_range, losses, 'orange', label='train loss') plt.plot(epochs_range, val_losses, '-.k', label='test loss') # plt.title('loss') plt.xlabel('Epochs') plt.ylabel('Loss') plt.legend() plt.savefig('experiments/loss_figure.png')

这是一个用于绘制训练和验证损失的函数。它接受三个参数：n_epochs 表示训练迭代的总数，losses 表示每个迭代的训练损失，val_losses 表示每个迭代的验证损失。它使用 matplotlib 库来创建一个简单的折线图，其中包含训练和验证损失。每个损失都用不同的颜色表示，并添加了图例。最后，它将图像保存到 experiments/loss_figure.png 文件中。