chrisgschon / gluon-NN-wrap

A wrapper for searching over Neural network architectures, built on top of MXNet framework, in particular gluon.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

gluon-NN-wrap

A wrapper for searching over neural network architectures, built on top of MXNet framework, in particular the gluon high level API.

Note that gluon_search.py's search_over_NNs expects the output labels to be two class 1/0s. Additional functions may be added in the future to explore multiclass/regression problems in a similar way.

About

A wrapper for searching over Neural network architectures, built on top of MXNet framework, in particular gluon.

License:MIT License


Languages

Language:Jupyter Notebook 55.6%Language:Python 44.4%