Skip to content
No description or website provided.
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
MNIST_data initial commit Dec 11, 2017
experiments Create README.md Dec 12, 2017
.DS_Store initial commit Dec 11, 2017
LICENSE Create LICENSE Dec 16, 2017
README.md Update README.md Dec 12, 2017
cnn.py initial commit Dec 11, 2017
net_manager.py initial commit Dec 11, 2017
reinforce.py initial commit Dec 11, 2017
train.py Fix bug of not importing datetime Apr 1, 2018

README.md

nascell-automl

This code belongs to the "Simple implementation of Neural Architecture Search with Reinforcement Learning " blog post.

An original blog post with all the details (step-by-step guide): https://lab.wallarm.com/the-first-step-by-step-guide-for-implementing-neural-architecture-search-with-reinforcement-99ade71b3d28

Requirements

  • Python 3
  • Tensorflow > 1.4

Training

Print parameters:

python3 train.py --help
optional arguments:
  -h, --help            show this help message and exit
  --max_layers MAX_LAYERS

Train:

python3 train.py

For evaluate architecture

Print parameters:

$ cd experiments/
$ python3 train.py --architecture "61, 24, 60,  5, 57, 55, 59, 3"
You can’t perform that action at this time.