API Reference
In this section, implemented classes are explained and their methods are described. In addition, the utils functions implemented in this package are described. For these classes and functions, the necessary and optional parameters are documented here.
The package consists of two subpackages with different purposes. The models folder contains different architectures of neural networks implemented in PyTorch, including different neural networks classes for predicting regression tasks or time series forecasting. It also includes a wrapper to build ensembles of models, where the outputs of multiple models are averaged.
The second subpackage contains utils functions. There are functions for data creation and formatting, but also for evaluating and investigating your trained model, including an analysis of the hidden-layer correlation, input/output sensitivity and visualization.
Models
Implemented models for regression tasks are:
If you have only a small number of training data or need explainability of the model, have a look at:
There are multiple neural networks (and their according cells) implemented for time series forecasting:
- Error Correction Neural Network
- Error Correction Neural Network Cell
- Error Correction Neural Network ECNN GRU 3 Variant
- Historical Consistent Neural Network
- Historical Consistent Neural Network Cell
- Historical Consistent Neural Network GRU Variant 3 Cell
- Deep Historical Consistent Neural Network
- Causal-Retro-Causal Neural Network
- Historical Consistent Neural Network With_Know_U
The Ensemble is applicable for all the above models and a lot more:
Utils
The utils subpackage contains functions to generate and prepare data for training:
The following utils are interesting for evaluating and investigating your neural network or visualize forecasts with or without uncertainty: