3. Designing our neural network

Now we are defining how many layers we have in our network along with its hidden units

n_input_unit = X_train.shape[1]
n_hidden_unit_1 = 4
n_output_unit = y_train.shape[1]
                              
print("Input layer: {} visible units".format(n_input_unit))
print("First layer: {} hidden units".format(n_hidden_unit_1))
print("Output layer: {} output unit ".format(n_output_unit))

Input layer: 2 visible units
First layer: 4 hidden units
Output layer: 1 output unit