日志保存--Python

日志保存–Python

本文方法来自：MINA: Multilevel Knowledge-Guided Attention for Modeling Electrocardiography Signals, IJCAI 2019一文的代码

训练模型的时候，经常需要保存中间训练过程的日志，可以采用以下方法：

run_id = 'mina_{0}'.format(strftime("%Y-%m-%d-%H-%M-%S", localtime()))
directory = 'res/{0}'.format(run_id)
# 判断路径是否存在，如果不存在则创建
try:
    os.stat('res/')
except:
    os.mkdir('res/')    
try:
    os.stat(directory)
except:
    os.mkdir(directory)  
log_file = '{0}/log.txt'.format(directory)
# 将代码拷贝到相应目录下
model_file = 'model_name.py'
copyfile(model_file, '{0}/{1}'.format(directory, model_file))
# 这种记录方式可以累积记录，下一次再使用时，不会覆盖前一次的记录
# run_id:  写入文件的内容
with open(log_file, 'a') as fout:
    print(run_id, file=fout)