← all repositories

titu1994/neural-architecture-search

A Keras/TensorFlow implementation of Neural Architecture Search with a Controller RNN that discovers optimal network architectures via reinforcement learning.

447 stars Python ML FrameworksAgents
neural-architecture-search
Velocity · 7d
+0.1
★ / day
Trend
steady
star history

This repository implements the Controller RNN from the NAS paper and NASNet paper, using an RL agent to search the space of neural network architectures. The Controller samples actions defining network parameters (kernel size, filters), while a NetworkManager trains generated Keras child networks and computes rewards. The implementation uses StateSpace to define the architecture search space and manages the training loop between controller and child networks.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.