在的线41MNIST例如,还必须在输出尺寸改变10到1network = fully_connected(network, 10, activation='softmax')
到network = fully_connected(network, 1, activation='linear')
。请注意,您可以删除最终的softmax。
查看您的代码,似乎您有一个目标值Y
,这意味着将 与 (您将在此处找到所有可用的损耗)一起使用:
regression(network, optimizer='adam', learning_rate=0.01,
loss='mean_square', name='target')
另外,将Y和Y_test整形为形状(batch_size,1)。