molecules.data.dataloaders package
Submodules
molecules.data.dataloaders.dataset module
-
class
molecules.data.dataloaders.dataset.
ContactMapDataset
(path, name=None)[source]
Bases: object
Data handling for Numpy stored contact maps
-
load_data
(shape=None)[source]
Load numpy array data.
Parameters: | shape (tuple, optional) – Shape of the data. Format: (H x W x C) |
Returns: |
- X_train (np.ndarray) – Training set.
- X_test (np.ndarray) – Test set.
|
molecules.data.dataloaders.fspeptide module
molecules.data.dataloaders.onefme module
molecules.data.dataloaders.utils module
Module contents