/gluon-NN-wrap

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

Primary LanguageJupyter NotebookMIT LicenseMIT

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.