Create Ann

ann

#initialize the ANN
classifier = Sequential()

#add input layer and first hidden layer
classifier.add(Dense(activation="relu", input_dim=11, units=6, kernel_initializer="uniform"))

#add second hidden layer
classifier.add(Dense(activation="relu", units=6, kernel_initializer="uniform"))

#output layer:
classifier.add(Dense(activation="sigmoid", units=1, kernel_initializer="uniform"))

#compile ANN
classifier.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])

#Fitting the ANN to data
classifier.fit(X_train, y_train, batch_size=10, epochs=100)