tf_geometric.datasets¶

Contents

Planetoid
Cora
Citeseer
Pubmed
SupervisedCora
SupervisedCiteseer
SupervisedPubmed
BlogCatalog
PPI
Reddit
TU
OGB
ModelNet
AmazonElectronics
Coauthor
Fraud Detection
HGB
NARS Academic

Planetoid ¶

class tf_geometric.datasets.PlanetoidDataset(dataset_name, task='semi_supervised', dataset_root_path=None)¶

__init__(dataset_name, task='semi_supervised', dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

Cora ¶

class tf_geometric.datasets.CoraDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

Citeseer ¶

class tf_geometric.datasets.CiteseerDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

Pubmed ¶

class tf_geometric.datasets.PubmedDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

SupervisedCora ¶

class tf_geometric.datasets.SupervisedCoraDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

SupervisedCiteseer ¶

class tf_geometric.datasets.SupervisedCiteseerDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

SupervisedPubmed ¶

class tf_geometric.datasets.SupervisedPubmedDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “cora” | “citeseer” | “pubmed”
task – “semi_supervised” | “supervised”
dataset_root_path –

BlogCatalog ¶

class tf_geometric.datasets.MultiLabelBlogCatalogDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

PPI ¶

class tf_geometric.datasets.PPIDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

Reddit ¶

class tf_geometric.datasets.TransductiveRedditDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

class tf_geometric.datasets.InductiveRedditDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

TU ¶

class tf_geometric.datasets.TUDataset(dataset_name, dataset_root_path=None)¶

__init__(dataset_name, dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

OGB ¶

class tf_geometric.datasets.OGBNodePropPredDataset(dataset_name, dataset_root_path=None)¶

OGB Node Property Prediction Datasets: https://ogb.stanford.edu/docs/nodeprop/

__init__(dataset_name, dataset_root_path=None)¶

OGB Node Property Prediction Datasets: https://ogb.stanford.edu/docs/nodeprop/

Parameters

dataset_name – “ogbn-arxiv” | “ogbn-products” | “ogbn-proteins” | “ogbn-papers100M” | “ogbn-mag”
dataset_root_path –

ModelNet ¶

class tf_geometric.datasets.ModelNet10Dataset(dataset_root_path=None, num_processes=50)¶

__init__(dataset_root_path=None, num_processes=50)¶: Initialize self. See help(type(self)) for accurate signature.

class tf_geometric.datasets.ModelNet40Dataset(dataset_root_path=None, num_processes=50)¶

__init__(dataset_root_path=None, num_processes=50)¶: Initialize self. See help(type(self)) for accurate signature.

AmazonElectronics ¶

class tf_geometric.datasets.AmazonElectronicsDataset(dataset_name, dataset_root_path=None)¶

__init__(dataset_name, dataset_root_path=None)¶

Parameters

dataset_name – “amazon-computers” | “amazon-photo”
dataset_root_path –

class tf_geometric.datasets.AmazonComputersDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “amazon-computers” | “amazon-photo”
dataset_root_path –

class tf_geometric.datasets.AmazonPhotoDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “amazon-computers” | “amazon-photo”
dataset_root_path –

Coauthor ¶

class tf_geometric.datasets.CoauthorDataset(dataset_name, dataset_root_path=None)¶

__init__(dataset_name, dataset_root_path=None)¶

Parameters

dataset_name – “coauthor-cs” | “coauthor-physics”
dataset_root_path –

class tf_geometric.datasets.CoauthorCSDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “coauthor-cs” | “coauthor-physics”
dataset_root_path –

class tf_geometric.datasets.CoauthorPhysicsDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “coauthor-cs” | “coauthor-physics”
dataset_root_path –

Fraud Detection ¶

class tf_geometric.datasets.FDAmazonDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

class tf_geometric.datasets.FDYelpChiDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶: Initialize self. See help(type(self)) for accurate signature.

HGB ¶

class tf_geometric.datasets.HGBDataset(dataset_name, dataset_root_path=None)¶

__init__(dataset_name, dataset_root_path=None)¶

Parameters

dataset_name – “hgb_acm” | “hgb_dblp” | “hgb_freebase” | “hgb_imdb”
dataset_root_path –

class tf_geometric.datasets.HGBACMDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “hgb_acm” | “hgb_dblp” | “hgb_freebase” | “hgb_imdb”
dataset_root_path –

class tf_geometric.datasets.HGBDBLPDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “hgb_acm” | “hgb_dblp” | “hgb_freebase” | “hgb_imdb”
dataset_root_path –

class tf_geometric.datasets.HGBFreebaseDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “hgb_acm” | “hgb_dblp” | “hgb_freebase” | “hgb_imdb”
dataset_root_path –

class tf_geometric.datasets.HGBIMDBDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “hgb_acm” | “hgb_dblp” | “hgb_freebase” | “hgb_imdb”
dataset_root_path –

NARS Academic ¶

class tf_geometric.datasets.NARSACMDataset(dataset_root_path=None)¶

__init__(dataset_root_path=None)¶

Parameters

dataset_name – “nars_academic_acm”
dataset_root_path –