当前位置:如说吧 > 休闲 > 正文

如何使用keras框架

2020-08-02 0

Keras输出的loss,val这些值保存到文本中去:
Keras中的fit函数回一个History对象,它的History.history属性之前些值全保存在里面,如果有验证集的话,也包含了验证集的这些指标变化情况,具体写法:
hist=model.fit(train_set_x,train_set_y,batch_size=256,shuffle=True,nb_epoch=nb_epoch,validation_split=0.1)with open(log_sgd_big_32.txt,w) as f:f.write(str(hist.history))

我觉得保存之前的loss,val这些值还是比较重要的,在之后的调参过程中有时候还是需要之前loss的结果作为参考的,特别是你自己添加了一些自己的loss的情况下,但是这样的写法会使整个文本的取名比较乱,所以其实可以考虑使用Aetros的插件。
本周热门
热门文章
热门关注