diff --git a/ann.py b/ann.py index a8a3d54..bc05d09 100644 --- a/ann.py +++ b/ann.py @@ -52,10 +52,10 @@ classifier = Sequential() # Adding the input layer and the first hidden layer -classifier.add(Dense(units = 6, kernel_initializer = 'uniform', activation = 'relu', input_dim = 11)) +classifier.add(Dense(units = 6, kernel_initializer = 'uniform', activation = 'tanh', 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'))