老卫带你学---keras画acc和loss曲线图

最新推荐文章于 2022-08-29 14:23:35 发布

老卫带你学

最新推荐文章于 2022-08-29 14:23:35 发布

阅读量1.8k

点赞数 1

分类专栏： keras

本文链接：https://blog.csdn.net/yixieling4397/article/details/114735951

版权

keras 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种在Keras中实现训练过程可视化的方法。通过定义一个回调函数LossHistory，可以记录训练过程中的loss和accuracy等指标，并使用matplotlib将这些指标绘制成图表，便于观察模型的训练情况。

摘要由CSDN通过智能技术生成

通过实现回调函数来实现，loss和acc的画图功能

from __future__ import print_function
import numpy as np


import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation
from keras.optimizers import SGD, Adam, RMSprop
from keras.utils import np_utils
import matplotlib.pyplot as plt


class LossHistory(keras.callbacks.Callback):
    def on_train_begin(self, logs={}):
        self.losses = {'batch':[], 'epoch':[]}
        self.accuracy = {'batch':[], 'epoch':[]}
        self.val_loss = {'batch':[], 'epoch':[]}
        self.val_acc = {'batch':[], 'epoch':[]}

    def on_batch_end(self, batch, logs={}):
        self.losses['batch'].append(logs.get('loss'))
        self.accuracy['batch'].append(logs.get('acc'))
        self.val_loss['batch'].append(logs.get('val_loss'))
        self.val_acc['batch'].append(logs.get('val_acc'))

    def on_epoch_end(self, batch, logs={}):
        self.losses['epoch'].append(logs.get('loss'))
        self.accuracy['epoch'].append(logs.get('acc'))
        self.val_loss['epoch'].append(logs.get('val_loss'))
        self.val_acc['epoch'].append(logs.get('val_acc'))

    def loss_plot(self, loss_type):
        iters = range(len(self.losses[loss_type]))
        plt.figure()
        # acc
        plt.plot(iters, self.accuracy[loss_type], 'r', label='train acc')
        # loss
        plt.plot(iters, self.losses[loss_type], 'g', label='train loss')
        if loss_type == 'epoch':
            # val_acc
            plt.plot(iters, self.val_acc[loss_type], 'b', label='val acc')
            # val_loss
            plt.plot(iters, self.val_loss[loss_type], 'k', label='val loss')
        plt.grid(True)
        plt.xlabel(loss_type)
        plt.ylabel('acc-loss')
        plt.legend(loc="upper right")
        plt.show()

紧接着生成这个类对象，并将其送入keras的回调函数中。

    history=losshistory.LossHistory()
    finetuned_model.fit_generator(batches, steps_per_epoch=num_train_steps, epochs=100, callbacks=[early_stopping, checkpointer,history], validation_data=val_batches, validation_steps=num_valid_steps)
    history.loss_plot('epoch')

在这里插入图片描述