keras模型训练过程绘制

        keras版本2.2.0,发现通过之前的history=model.fit(),print(history.history())会出现一个问题'dict' object is not callable然后找了中文官方文档以及其他网上资料,依然没有解决,最后去看了下外文的官方文档https://keras.io/callbacks/#history。在外文官方文档中有这么一个例子:


class LossHistory(keras.callbacks.Callback):
    def on_train_begin(self, logs={}):
        self.losses = []

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

model = Sequential()
model.add(Dense(10, input_dim=784, kernel_initializer='uniform'))
model.add(Activation('softmax'))
model.compile(loss='categorical_crossentropy', optimizer='rmsprop')

history = LossHistory()
model.fit(x_train, y_train, batch_size=128, epochs=20, verbose=0, callbacks=[history])

print(history.losses)
# outputs
'''
[0.66047596406559383, 0.3547245744908703, ..., 0.25953155204159617, 0.25901699725311789]
'''

        然后根据这个例子在此基础上进行修改,发现可以打印出来训练过程中的参数。

from keras.callbacks import Callback
class History(Callback):
    def on_train_begin(self,logs={}):
        self.losses=[]
        self.acces=[]
    def on_batch_end(self,batch,logs={}):
        self.losses.append(logs.get('loss'))
        self.acces.append(logs.get('acc'))
history=History()
"""model这儿换上自己定义的模型就行"""
model.fit(X_train,y_train,batch_size=35,epochs=2,verbose=1,validation_data=(X_val,y_val),callbacks=[history])
print(history.losses)
print(history.access)
"""自己可以再重新定义一个数组用来转每个步长训练过程中的数据,让所有过程数据存放在一个数组中,最后通过matplot绘制"""

 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值