如何在keras中存储每个epoch摘要变量?

2024-04-20 07:02:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试在python上用keras为卷积神经网络创建GUI。 我想知道如何保存每个摘要model.fit_发电机(…)每次在一个epoch完成时,都要在一个变量中输入一个变量(这样我就可以把它放在一个显示框中;每当一个epoch完成显示框时,或者变量都会用新的keras输出进行更新)。在

输出参考: 我说的是每次都要保存在变量中的以下行。在

0s - loss: 0.6551 - acc: 0.5000 - val_loss: 0.9836 - val_acc: 0.4000

Tags: modelgui神经网络val卷积发电机fitkeras
1条回答
网友
1楼 · 发布于 2024-04-20 07:02:26

你应该创建一个回调函数来更新你的“显示框”。 下面是一个在每个历元之后打印所有值的示例。你可以很容易地改变它以满足你的需要。在

class My_Callback(keras.callbacks.Callback):
    def on_train_begin(self, logs={}):
        return

    def on_train_end(self, logs={}):
        return

    def on_epoch_begin(self, logs={}):
        return

    def on_epoch_end(self, epoch, logs={}):
        def on_epoch_end(self, epoch, logs=None):
        logs = logs or {}
        for k in self.params['metrics']:
            if k in logs:
                print("Name: %s, Value: %s" % (k, logs[k]))
    def on_batch_begin(self, batch, logs={}):
        return

    def on_batch_end(self, batch, logs={}):
        self.losses.append(logs.get('loss'))
        return

您还应该查看处理您所引用的stdout输出的ProbarLogger()回调: https://github.com/fchollet/keras/blob/master/keras/callbacks.py#L245

相关问题 更多 >