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