Skip to content

Instantly share code, notes, and snippets.

@97k

97k/architecture.py

Last active March 18, 2019 09:29
Show Gist options
  • Save 97k/7dee9d2616bcce33b7feaf624af8081a to your computer and use it in GitHub Desktop.
Save 97k/7dee9d2616bcce33b7feaf624af8081a to your computer and use it in GitHub Desktop.
Download ZIP
neural network architecture
Raw
architecture.py
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
]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment