diff --git a/ann.py b/ann.py index a8a3d54..1f01402 100644 --- a/ann.py +++ b/ann.py @@ -55,7 +55,7 @@ classifier.add(Dense(units = 6, kernel_initializer = 'uniform', activation = 'relu', input_dim = 11)) # Adding the second hidden layer -classifier.add(Dense(units = 6, kernel_initializer = 'uniform', activation = 'relu')) +classifier.add(Dense(units = 6, kernel_initializer = 'uniform', activation = 'tanh')) # Adding the output layer classifier.add(Dense(units = 1, kernel_initializer = 'uniform', activation = 'sigmoid'))