使用TensorFlow搭建我的第一个神经网络

训练集:有从0到5的数字的1080张图片(64x64像素),每个数字拥有180张图片。
测试集:有从0到5的数字的120张图片(64x64像素),每个数字拥有5张图片。
训练集样本数 = 1080
测试集样本数 = 120
X_train.shape: (12288, 1080)
Y_train.shape: (6, 1080)
X_test.shape: (12288, 120)
Y_test.shape: (6, 120)# 欢迎使用Markdown编辑器
在这里插入图片描述

实现一个三层的TensorFlow神经网络:LINEAR -> RELU -> LINEAR -> RELU -> LINEAR -> SOFTMAX 25----------------->12-------------------------->6 epoch表示迭代次数 最终结果 epoch = 0 epoch_cost = 1.85570192698 epoch = 100 epoch_cost = 1.01645786473 epoch = 200 epoch_cost = 0.733102418257 epoch = 300 epoch_cost = 0.572938044866 epoch = 400 epoch_cost = 0.468798786853 epoch = 500 epoch_cost = 0.380979135633 epoch = 600 epoch_cost = 0.313818839463 epoch = 700 epoch_cost = 0.254258293997 epoch = 800 epoch_cost = 0.203794926179 epoch = 900 epoch_cost = 0.166409768164 epoch = 1000 epoch_cost = 0.141497232923 epoch = 1100 epoch_cost = 0.10757850675 epoch = 1200 epoch_cost = 0.0862285523026 epoch = 1300 epoch_cost = 0.0594153394076 参数已经保存到session。 训练集的准确率: 0.998148 测试集的准确率: 0.758333 CPU的执行时间 = 701.1643028076872 秒