Skip to content

Instantly share code, notes, and snippets.

@SkalskiP
Created October 3, 2018 22:51
Show Gist options
  • Select an option

  • Save SkalskiP/7694859d2c4361bc31d30ce18e7d22d8 to your computer and use it in GitHub Desktop.

Select an option

Save SkalskiP/7694859d2c4361bc31d30ce18e7d22d8 to your computer and use it in GitHub Desktop.
Example of neural network architecture
nn_architecture = [
{"input_dim": 2, "output_dim": 4, "activation": "relu"},
{"input_dim": 4, "output_dim": 6, "activation": "relu"},
{"input_dim": 6, "output_dim": 6, "activation": "relu"},
{"input_dim": 6, "output_dim": 4, "activation": "relu"},
{"input_dim": 4, "output_dim": 1, "activation": "sigmoid"},
]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment