pyproteonet.dgl.collate.masked_heterograph_to_homogeneous

pyproteonet.dgl.collate.masked_heterograph_to_homogeneous(masked_heterographs: List[DGLGraph], target: str, features: List[str] = []) → List[DGLGraph]

Converts a list of DGL heterographs created from masked datasets to homogeneous graphs.

Parameters:

  • masked_heterographs (List[dgl.DGLGraph]) – A list of DGL heterographs.

  • target (str) – The target attribute name.

  • features (List[str], optional) – A list of feature attribute names. Defaults to [].

Returns:

A list of homogeneous graphs.

Return type:

List[dgl.DGLGraph]