【发布时间】:2018-06-27 23:55:51
【问题描述】:
我有一个关于如何定义神经网络的基本初学者问题,我正在 Keras 库的背景下学习。按照 MNIST hello world 程序,我定义了这个网络:
model = Sequential()
model.add(Dense(NB_CLASSES, input_shape=(RESHAPED,), activation='softmax'))
我的理解是,这会创建一个有两层的神经网络,在这种情况下,RESHAPED 是 784,NB_CLASSES 是 10,所以网络将有 1 个输入层,有 785 个神经元,一个输出层有 10 个神经元神经元。
然后我添加了这个:
model.compile(loss='categorical_crossentropy', optimizer=OPTIMIZER, metrics=['accuracy'])
我知道已经阅读了分类交叉熵的公式,但它似乎是按输出节点计算的。我的问题是,在训练期间,如何组合交叉熵的值来创建标量值目标函数?这只是一个平均值吗?
【问题讨论】:
标签: neural-network keras keras-layer loss-function cross-entropy