Skip to content
/ DNNs Public

A package for implemnting Robust Neural Network with Logging models, Ensembles, Early Stopping, and much more.

Notifications You must be signed in to change notification settings

parthatom/DNNs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

41 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DNNs

A repository for easy training of Neural Networks.

Present Features:

  • Ability to declare and train models with less than 5 lines of code.
  • Define dropout rates, batch norms, and more such features easily.
  • Logging enabled - researchers can log their metrics in a clean pandas file without having to write any regex expressions and such.
  • Use along with a transformer: A transformer (more precisely data modifier), can be anything from a pre-trained model to a data preparation pipeline. Additional support for pytorch models is added.
  • Create ensembles with ease through ensemble.py

About

A package for implemnting Robust Neural Network with Logging models, Ensembles, Early Stopping, and much more.

Topics

Resources

Stars

Watchers

Forks

Languages