阅读( 141 )次

   评论(0)条

wudixx

学习笔记之第3章:用于MNIST的卷积神经网络 / 第3节:简单卷积神经网络的训练和评估会话

1.整体思路


2.运行后在excel中查看结果


3.保存评估结果

(1)导入scv

import csv

(2)建立list列表

#将评估结果保存到文件
results_list = list()
# 写入参数配置(studyai.com)
results_list.append(['learning_rate', learning_rate_init,
'training_epochs', training_epochs,
'batch_size', batch_size,
'display_step', display_step])
results_list.append(['train_step', 'train_loss', 'validation_loss',
'train_step', 'train_accuracy', 'validation_accuracy'])

# 将评估结果保存到文件(studyai.com)
results_list.append([training_step, train_loss, validation_loss,
training_step, train_acc, validation_acc])


(3)保存文件

# 将评估结果保存到文件
results_file = open('evaluate_results/evaluate_results.csv', 'w', newline='')
csv_writer = csv.writer(results_file, dialect='excel')
for row in results_list:
csv_writer.writerow(row)


 收藏 (0)  打赏  点赞 (0)

 ©2017 studyai.com 版权所有

关于我们