diff --git a/ann.py b/ann.py index a8a3d54..90d409b 100644 --- a/ann.py +++ b/ann.py @@ -52,13 +52,13 @@ 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')) +classifier.add(Dense(units = 1, kernel_initializer = 'uniform', activation = 'relu')) # Compiling the ANN classifier.compile(optimizer = 'adam', loss = 'binary_crossentropy', metrics = ['accuracy'])