Framework for the automatic creation of CNN architectures
Project description
TorchCNNBuilder
TorchCNNBuilder is an open-source framework for the automatic creation of CNN architectures. This framework should first of all help researchers in the applicability of CNN models for a huge range of tasks, taking over most of the writing of the architecture code. This framework is distributed under the 3-Clause BSD license. All the functionality is written only using pytorch
(no third-party dependencies)
Installation
The simplest way to install framework is using pip
:
pip install torchcnnbuilder
Usage examples
The basic structure of the framework is presented below. Each subdirectory has its own example of using the appropriate available functionality. You can check <directory>_examples.ipynb
files in order to see the ways to use the proposed toolkit. In short, there is the following functionality:
- the ability to calculate the size of tensors after (transposed) convolutional layers
- preprocessing an n-dimensional time series in
TensorDataset
- automatic creation of (transposed) convolutional sequences
- automatic creation of (transposed) convolutional layers and (transposed) blocks from convolutional layers
The structure of the main part of the package:
├── examples
│ ├── builder_examples.ipynb
│ ├── preprocess_examples.ipynb
│ ├── models_examples.ipynb
│ └── tools # additional functions for the examples
└── torchcnnbuilder
├── preprocess
│ └── time_series.py
├── builder.py
└── models.py
Initially, the library was created to help predict n-dimensional time series (geodata), so there is a corresponding functionality and templates of predictive models (like ForecasterBase
)
Sources
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for torchcnnbuilder-0.0.16-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3a18438a68cb61587a16c34ce17c30cc7b8d7bc1abd35637c36de371e6952e08 |
|
MD5 | 06c2feceb6a1cc60339c7a7f9d43f70e |
|
BLAKE2b-256 | 806eb2ec60fa6ea6bb95ddeea32387533e01211fdfeb4acaecf4dc17dd857e18 |