alazareva / ECNN

Automatic CNN generator

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ECNN

This package uses adaptive evolutionary algorithms in order to train, evaluate, and evolve Convolutional Neural Networks. It includes a simple interface with TensorFlow and allows for parallel model training across multiple GPUs.

Getting Started

The provided code uses the CIFAR 10 dataset as input. The can be downloaded here. The dataset should be saved in the 'datasets/cifar10' directory. Other datasets can be adapted to use the DataSet class provided in dataset.py.

Tournament defautls can be set in defaults.py.

Prerequisities

  • Python 3
  • Numpy
  • TensorFlow
  • namedlist
  • Pickle

Examples

For full list of results please see the Report. Example tournament results showing mutations applied at each generation and resulting CNN performance.

Ex1 Ex2

Authors

  • Anastasiya Lazareva - Initial work - alazareva

About

Automatic CNN generator


Languages

Language:Python 100.0%