terminator.data.data

Datasets and dataloaders for loading TERMs.

This file contains dataset and dataloader classes to be used when interacting with TERMs.

Functions

convert(tensor)	Converts given tensor from numpy to pytorch.
load_file(in_folder, pdb_id[, min_protein_len])	Load the data specified in the proper .features file and return them.
read_lens(in_folder, pdb_id[, min_protein_len])	Reads the lengths specified in the proper .length file and return them.

Classes

TERMBatchSampler(dataset[, batch_size, ...])	BatchSampler/Dataloader helper class for TERM data using TERMDataset.
TERMDataset(in_folder[, pdb_ids, ...])	TERM Dataset that loads all feature files into a Pytorch Dataset-like structure.
TERMLazyBatchSampler(dataset[, batch_size, ...])	BatchSampler/Dataloader helper class for TERM data using TERMLazyDataset.
TERMLazyDataset(in_folder[, pdb_ids, ...])	TERM Dataset that loads all feature files into a Pytorch Dataset-like structure.