这是在TensorFlow中进行自定义估算器的指南: https://www.tensorflow.org/guide/custom_estimators
隐藏层是使用tf.nn.relu
制成的:
# Build the hidden layers, sized according to the 'hidden_units' param.
for units in params['hidden_units']:
net = tf.layers.dense(net, units=units, activation=tf.nn.relu)
我使用hidden_units=[4]
和n_classes=2
对示例进行了一些改动以学习 XOR 。当激活功能更改为tf.nn.sigmoid
时,该示例照常工作。 为什么会这样?因为XOR输入只是零和一,它是否仍能给出正确的结果?
这两个函数都可使平滑的损耗曲线收敛到零线。