wb = xlwt.Workbook()
#生成方法
train = Train(config)
train.remove_logdir()
#得到列表型数据
(acc_list,loss_list,vali_acc_list,vali_loss_list,lr_list)=train.start_train()
#写入excel文件
# 对acc和loss记录到excel中,方便用其他软件绘图
sheet = wb.add_sheet("Sheet1")
head_data = ["acc", "loss", "val_acc", "val_loss","lr"]
#print(acc_list)
# write函数写入,分别是write(x行 x列 数据)
for head in head_data:
sheet.write(0, head_data.index(head), head)
for i in range(len(acc_list)):
print(i+1)
sheet.write(i+1, 0, str(acc_list[i]))
sheet.write(i+1, 1, str(loss_list[i]))
sheet.write(i+1, 2, str(vali_acc_list[i]))
sheet.write(i+1, 3, str(vali_loss_list[i]))
sheet.write(i+1, 4, str(lr_list[i]))
wb.save("acc_loss.xls")
python-使用xlwt把列表数据写入excel指定位置
最新推荐文章于 2023-09-21 12:28:49 发布