Browse Source

NeuralNetXor: Changed activation to tanh

tags/v0.12
Eli Belash 6 years ago
parent
commit
c7bc4dd19e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test/TensorFlowNET.Examples/BasicModels/NeuralNetXor.cs

+ 1
- 1
test/TensorFlowNET.Examples/BasicModels/NeuralNetXor.cs View File

@@ -54,7 +54,7 @@ namespace TensorFlowNET.Examples
var logits = tf.matmul(hidden_activations, output_weights);

// Shape [4]
var predictions = tf.sigmoid(tf.squeeze(logits));
var predictions = tf.tanh(tf.squeeze(logits));
var loss = tf.reduce_mean(tf.square(predictions - tf.cast(labels, tf.float32)), name:"loss");

var gs = tf.Variable(0, trainable: false, name: "global_step");


Loading…
Cancel
Save