Commit 02d1db1c authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

changed default activation function to tanh

parent 0ba9f83e
......@@ -45,7 +45,7 @@ class AdaNetWrapper(object):
self.adanet_lambda = kwargs.get("adanet_lambda", 0.001)
self.random_seed = int(kwargs.get("random_seed", 42))
self.model_dir = kwargs.get("model_dir", None)
self.activation = kwargs.get("activation", tf.nn.relu)
self.activation = kwargs.get("activation", tf.nn.tanh)
self.shuffles = int(kwargs.get("shuffles", 10))
self.dropout_rate = float(kwargs.get("dropout_rate", 0.2))
self.augmentation = kwargs.get("augmentation", False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment