これは、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)
と を使用してXORを学習するために、例を少し変更しました。アクティベーション関数を に変更すると、例は通常どおりに機能します。なぜそうなのですか?XOR 入力は 0 と 1 だけなので、正しい結果が得られますか?hidden_units=[4]
n_classes=2
tf.nn.sigmoid
どちらの関数も、ゼロ ラインに収束する滑らかな損失曲線を提供します。