Welcome to nn-toolbox’s documentation!¶
Implementations of training procedures |
|
General components for neural networks |
|
Methods for generating an ensemble of models |
|
Abstraction for pytorch hooks (similar to callbacks but for layers) |
|
Initializers for weight layers |
|
Some generic loss functions |
|
Metrics for model evaluation |
|
Abstraction for machine learning modelling (e.g classifier, ensemble, etc.) |
|
Optimizers and optimization utility functions |
|
Components, learners, losses, models, transformations and utility functions for sequence tasks. |
|
For tabular data and recommendation systems |
|
Some generic data transformations |
|
Some generic utilities function for neural networks |
|
Components, learners, losses, models, transformations and utility functions for vision tasks. |
|
Utilities for model and data visualization |