cogdl.datasets.pyg

Module Contents

Functions

normalize_feature(data)

cogdl.datasets.pyg.normalize_feature(data)[source]
class cogdl.datasets.pyg.CoraDataset[source]

Bases: torch_geometric.datasets.Planetoid

class cogdl.datasets.pyg.CiteSeerDataset[source]

Bases: torch_geometric.datasets.Planetoid

class cogdl.datasets.pyg.PubMedDataset[source]

Bases: torch_geometric.datasets.Planetoid

class cogdl.datasets.pyg.RedditDataset[source]

Bases: torch_geometric.datasets.Reddit

class cogdl.datasets.pyg.MUTAGDataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.ImdbBinaryDataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.ImdbMultiDataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.CollabDataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.ProtainsDataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.RedditBinary[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.RedditMulti5K[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.RedditMulti12K[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.PTCMRDataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.NCT1Dataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.NCT109Dataset[source]

Bases: torch_geometric.datasets.TUDataset

class cogdl.datasets.pyg.ENZYMES[source]

Bases: torch_geometric.datasets.TUDataset

__getitem__(self, idx)[source]
class cogdl.datasets.pyg.QM9Dataset[source]

Bases: torch_geometric.datasets.QM9