NN_ARCHITECTURE = [ {"input_dim": 3, "output_dim": 4, "activation": "relu"},# First Hidden Layer {"input_dim": 4, "output_dim": 4, "activation": "relu"},# Second Hidden Layer {"input_dim": 4, "output_dim": 1, "activation": "sigmoid"},# Output Layer ]