pyproteonet.dgl.collate.masked_dataset_to_homogeneous_graph

pyproteonet.dgl.collate.masked_dataset_to_homogeneous_graph(masked_datasets: List[MaskedDataset], mappings: List[str], target: str, features: List[str] = [], sample_lists: List[str] | None = None) → List[DGLGraph]

Converts a list of masked datasets into a list of homogeneous DGL graphs.

Parameters:

• masked_datasets (List[MaskedDataset]) – List of masked datasets.

• mappings (List[str]) – List of mappings.

• target (str) – Target column.

• features (List[str], optional) – List of feature columns. Defaults to [].

• sample_lists (Optional[List[str]], optional) – List of sample names. Defaults to None.

Returns:

Homogeneous graph representation.

Return type:

dgl.DGLGraph