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

Revisions

  1. 97k revised this gist Mar 18, 2019. 1 changed file with 5 additions and 6 deletions.
    11 changes: 5 additions & 6 deletions architecture.py
    View file
    Original file line number Diff line number Diff line change
    @@ -1,7 +1,6 @@
    NN_ARCHITECTURE = [
    {"input_dim": 2, "output_dim": 32, "activation": "relu"}, # Input Layer
    {"input_dim": 32, "output_dim": 64, "activation": "relu"},# First Hidden Layer
    {"input_dim": 64, "output_dim": 64, "activation": "relu"},# Second Hidden Layer
    {"input_dim": 64, "output_dim": 32, "activation": "relu"},# Third Hidden Layer
    {"input_dim": 32, "output_dim": 1, "activation": "sigmoid"},# Output Layer
    ]
    {"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
    ]

  2. 97k created this gist Mar 17, 2019.
    7 changes: 7 additions & 0 deletions architecture.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,7 @@
    NN_ARCHITECTURE = [
    {"input_dim": 2, "output_dim": 32, "activation": "relu"}, # Input Layer
    {"input_dim": 32, "output_dim": 64, "activation": "relu"},# First Hidden Layer
    {"input_dim": 64, "output_dim": 64, "activation": "relu"},# Second Hidden Layer
    {"input_dim": 64, "output_dim": 32, "activation": "relu"},# Third Hidden Layer
    {"input_dim": 32, "output_dim": 1, "activation": "sigmoid"},# Output Layer
    ]