Training wrapper around torch.nn.Module, provides scikit-learn like fit and predict interfaces.
Project description
torch-train
Training wrapper around torch.nn.Module, provides scikit-learn like fit and predict interfaces.
Installation
The easiest way to install is through pip
pip install torch-train
Dependencies
The torch-train is an extension of the pytorch library. Therefore, when installed manually, ensure pytorch is installed. Either through pip:
pip install torch
See https://pytorch.org/ for a detailed installation guide.
Documentation
For a full reference, please see torch-train.readthedocs.io.
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
torch-train-0.0.3.tar.gz
(4.1 kB
view hashes)
Built Distribution
Close
Hashes for torch_train-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 42ab3c4ba37f37b115723e6fdedcb6aaac6ae8ef1102d5675e6391ff5eff2916 |
|
MD5 | ce2a2c920a859770b77c2e1f161e0033 |
|
BLAKE2b-256 | bc9eab128cf91795cd82f8448eb791e23dbf0c1e1cb5bfaabc8482fe0c0932a1 |