tf_geometric.datasets
Contents
Reddit
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
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 –
-
Coauthor
- Parameters
dataset_name – “coauthor-cs” | “coauthor-physics”
dataset_root_path –
- Parameters
dataset_name – “coauthor-cs” | “coauthor-physics”
dataset_root_path –
- Parameters
dataset_name – “coauthor-cs” | “coauthor-physics”
dataset_root_path –
Fraud Detection
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 –
-