stcrpy.tcr_datasets package

Submodules

stcrpy.tcr_datasets.tcr_graph_dataset module

class stcrpy.tcr_datasets.tcr_graph_dataset.TCRGraphConstructor(config=None, **kwargs)[source]

Bases: object

build_graph(tcr: TCR, label=None)[source]

set_edge_featuriser(edge_featuriser_function, test_input=None)[source]

set_node_featuriser(node_featuriser_function, test_input=None)[source]

set_node_selector(node_selector_function)[source]

class stcrpy.tcr_datasets.tcr_graph_dataset.TCRGraphDataset(root, data_paths, graph_config=None, *args, **kwargs)[source]

Bases: Dataset

get(idx)[source]: Gets the data object at index idx.

len()[source]: Returns the number of data objects stored in the dataset.

pop(idx)[source]

process()[source]: Processes the dataset to the self.processed_dir folder.

property processed_file_names: The name of the files in the self.processed_dir folder that must be present in order to skip processing.

property raw_file_names: The name of the files in the self.raw_dir folder that must be present in order to skip downloading.

set_y(idx, label)[source]

stcrpy.tcr_datasets.tcr_selector module

stcrpy.tcr_datasets.tcr_structure_dataset module

stcrpy.tcr_datasets.utils module

Module contents