Getting Started
Core Modules
Aggregation
Aggregation.__init__()
Aggregation.execute()
Aggregation.reset_parameters()
Aggregation.assert_index_present()
Aggregation.assert_sorted_index()
Aggregation.assert_two_dimensional_input()
Aggregation.reduce()
Aggregation.segment()
Aggregation.to_dense_batch()
Aggregation.to_dense_batch_helper()
MultiAggregation
MultiAggregation.__init__()
MultiAggregation.reset_parameters()
MultiAggregation.get_out_channels()
MultiAggregation.execute()
MultiAggregation.combine()
SumAggregation
SumAggregation.execute()
MeanAggregation
MeanAggregation.execute()
MaxAggregation
MaxAggregation.execute()
MinAggregation
MinAggregation.execute()
MessagePassing
MessagePassing.special_args
MessagePassing.__init__()
MessagePassing.propagate()
MessagePassing.message()
MessagePassing.aggregate()
MessagePassing.update()
GCNConv
GCNConv.__init__()
GCNConv.reset_parameters()
GCNConv.execute()
GCNConv.propagate_msg()
GCNConv.propagate_spmm()
ChebConv
ChebConv.__init__()
ChebConv.reset_parameters()
ChebConv.execute()
ChebConv.forward()
ChebConv.message()
SGConv
SGConv.__init__()
SGConv.reset_parameters()
SGConv.execute()
SGConv.propagate_msg()
SGConv.propagate_spmm()
GCN2Conv
GCN2Conv.__init__()
GCN2Conv.reset_parameters()
GCN2Conv.execute()
GCN2Conv.propagate_msg()
GCN2Conv.propagate_spmm()
MessagePassingNts
MessagePassingNts.special_args
MessagePassingNts.__init__()
MessagePassingNts.propagate()
MessagePassingNts.aggregate_with_weight()
MessagePassingNts.scatter_to_edge()
MessagePassingNts.edge_forward()
MessagePassingNts.scatter_to_vertex()
MessagePassingNts.vertex_forward()
GATConv
GATConv.__init__()
GATConv.reset_parameters()
GATConv.execute()
GATConv.propagate()
GATConv.scatter_to_edge()
GATConv.edge_forward()
GATConv.scatter_to_vertex()
GATConv.vertex_forward()
EGNNConv
EGNNConv.__init__()
EGNNConv.init_weights()
EGNNConv.execute()
EGNNConv.message()
EGNNConv.propagate()
APPNP
APPNP.__init__()
APPNP.reset_parameters()
APPNP.execute()
APPNP.propagate_msg()
APPNP.propagate_spmm()
GPRGNN
GPRGNN.__init__()
GPRGNN.reset_parameters()
GPRGNN.execute()
GPRGNN.propagate_msg()
GPRGNN.propagate_spmm()
EvenNet
EvenNet.__init__()
EvenNet.reset_parameters()
EvenNet.execute()
EvenNet.propagate_msg()
EvenNet.propagate_spmm()
BernNet
BernNet.__init__()
BernNet.reset_parameters()
BernNet.execute()
BernNet.propagate_msg()
BernNet.propagate_spmm()
ChebNetII
ChebNetII.__init__()
ChebNetII.reset_parameters()
ChebNetII.execute()
ChebNetII.propagate_msg()
ChebNetII.propagate_spmm()
TransformerConv
TransformerConv.__init__()
TransformerConv.execute()
TransformerConv.forward()
TransformerConv.message()
TransformerConv.reset_parameters()
OptBasisConv
OptBasisConv.__init__()
OptBasisConv.reset_parameters()
OptBasisConv.three_term_prop()
OptBasisConv.execute()
OptBasisConv.propagate_msg()
OptBasisConv.propagate_spmm()
ClusterGCNConv
ClusterGCNConv.__init__()
ClusterGCNConv.reset_parameters()
ClusterGCNConv.execute()
ClusterGCNConv.propagate_msg()
ClusterGCNConv.propagate_spmm()
SAGEConv
SAGEConv.__init__()
SAGEConv.reset_parameters()
SAGEConv.execute()
SAGEConv.propagate_msg()
SAGEConv.propagate_spmm()
MergeLayer
MergeLayer.__init__()
MergeLayer.execute()
TimeEncoder
TimeEncoder.__init__()
TimeEncoder.execute()
MLP
MLP.__init__()
MLP.execute()
Data
Data.__init__()
Data.from_dict()
Data.to_dict()
Data.to_namedtuple()
Data.__getitem__()
Data.__setitem__()
Data.keys
Data.__len__()
Data.__contains__()
Data.__iter__()
Data.__call__()
Data.__cat_dim__()
Data.__inc__()
Data.num_nodes
Data.num_edges
Data.num_faces
Data.num_node_features
Data.num_features
Data.num_edge_features
Data.contains_isolated_nodes()
Data.contains_self_loops()
Data.apply()
Data.contiguous()
Data.cpu()
Data.cuda()
Data.clone()
Data.pin_memory()
Data.debug()
Dataset
Dataset.raw_file_names
Dataset.processed_file_names
Dataset.download()
Dataset.process()
Dataset.len()
Dataset.get()
Dataset.__init__()
Dataset.indices()
Dataset.raw_dir
Dataset.processed_dir
Dataset.num_node_features
Dataset.num_features
Dataset.num_edge_features
Dataset.raw_paths
Dataset.processed_paths
Dataset.__len__()
Dataset.__getitem__()
Dataset.index_select()
Dataset.shuffle()
InMemoryDataset
InMemoryDataset.raw_file_names
InMemoryDataset.processed_file_names
InMemoryDataset.download()
InMemoryDataset.process()
InMemoryDataset.__init__()
InMemoryDataset.num_classes
InMemoryDataset.len()
InMemoryDataset.get()
InMemoryDataset.collate()
InMemoryDataset.copy()
download_url()
decide_download()
extract_zip()
extract_gz()
extract_tar()
CSC
CSC.__init__()
CSR
CSR.__init__()
GraphChunk
GraphChunk.__init__()
GraphChunk.set_csr()
GraphChunk.save()
GraphChunk.load()
TemporalData
TemporalData.__init__()
TemporalData.__setitem__()
TemporalData.keys
TemporalData.num_nodes
TemporalData.max_node_id
TemporalData.src_size
TemporalData.dst_size
TemporalData.num_events
TemporalData.num_edges
TemporalData.apply()
TemporalData.to()
TemporalData.train_val_test_split()
TemporalData.train_val_test_split_w_mask()
TemporalData.seq_batches()
Batch
Batch.from_data_list()
Batch.get_example()
Batch.to_data_list()
Batch.num_graphs
Batch.batch_size
Dictionary
Dictionary.__init__()
Dictionary.__len__()
Dictionary.vec_index()
Dictionary.index()
Dictionary.special_index()
Dictionary.add_symbol()
Dictionary.bos()
Dictionary.pad()
Dictionary.eos()
Dictionary.unk()
Dictionary.load()
Dictionary.add_from_file()
ConformerGen
ConformerGen.__init__()
ConformerGen.transform_raw()
Planetoid
Planetoid.url
Planetoid.__init__()
Planetoid.raw_dir
Planetoid.processed_dir
Planetoid.raw_file_names
Planetoid.processed_file_names
Planetoid.download()
Planetoid.process()
Planetoid.make_link_split()
Amazon
Amazon.url
Amazon.__init__()
Amazon.raw_dir
Amazon.processed_dir
Amazon.raw_file_names
Amazon.processed_file_names
Amazon.download()
Amazon.process()
Amazon.make_link_split()
WikipediaNetwork
WikipediaNetwork.url
WikipediaNetwork.__init__()
WikipediaNetwork.raw_dir
WikipediaNetwork.processed_dir
WikipediaNetwork.raw_file_names
WikipediaNetwork.processed_file_names
WikipediaNetwork.download()
WikipediaNetwork.process()
GeomGCN
GeomGCN.url
GeomGCN.__init__()
GeomGCN.raw_dir
GeomGCN.processed_dir
GeomGCN.raw_file_names
GeomGCN.processed_file_names
GeomGCN.download()
GeomGCN.process()
LINKXDataset
LINKXDataset.__init__()
LINKXDataset.raw_dir
LINKXDataset.processed_dir
LINKXDataset.raw_file_names
LINKXDataset.processed_file_names
LINKXDataset.download()
LINKXDataset.process()
OGBNodePropPredDataset
OGBNodePropPredDataset.__init__()
OGBNodePropPredDataset.get_idx_split()
OGBNodePropPredDataset.num_classes
OGBNodePropPredDataset.raw_file_names
OGBNodePropPredDataset.processed_file_names
OGBNodePropPredDataset.download()
OGBNodePropPredDataset.process()
HeteroDataset
HeteroDataset.url
HeteroDataset.__init__()
HeteroDataset.raw_dir
HeteroDataset.processed_dir
HeteroDataset.raw_file_names
HeteroDataset.processed_file_names
HeteroDataset.download()
HeteroDataset.process()
JODIEDataset
JODIEDataset.url
JODIEDataset.names
JODIEDataset.__init__()
JODIEDataset.raw_dir
JODIEDataset.processed_dir
JODIEDataset.raw_file_names
JODIEDataset.processed_file_names
JODIEDataset.download()
JODIEDataset.process()
Reddit
Reddit.url
Reddit.__init__()
Reddit.raw_file_names
Reddit.processed_file_names
Reddit.download()
Reddit.process()
TemporalDataLoader
TemporalDataLoader.__init__()
QM9
QM9.raw_url
QM9.raw_url2
QM9.__init__()
QM9.mean()
QM9.std()
QM9.atomref()
QM9.raw_file_names
QM9.processed_file_names
QM9.download()
QM9.process()
QM9.get_idx_split()
MoleculeNet
MoleculeNet.url
MoleculeNet.names
MoleculeNet.__init__()
MoleculeNet.raw_dir
MoleculeNet.processed_dir
MoleculeNet.raw_file_names
MoleculeNet.processed_file_names
MoleculeNet.download()
MoleculeNet.process()
MD17
MD17.gdml_url
MD17.revised_url
MD17.file_names
MD17.__init__()
MD17.mean()
MD17.raw_dir
MD17.processed_dir
MD17.raw_file_names
MD17.processed_file_names
MD17.download()
MD17.process()
PCQM4Mv2
PCQM4Mv2.url
PCQM4Mv2.split_mapping
PCQM4Mv2.__init__()
PCQM4Mv2.raw_file_names
PCQM4Mv2.processed_file_names
PCQM4Mv2.download()
PCQM4Mv2.process()
MovieLens1M
MovieLens1M.url
MovieLens1M.__init__()
MovieLens1M.raw_file_names
MovieLens1M.download()
MovieLens1M.read_raw()
MovieLens100K
MovieLens100K.url
MovieLens100K.__init__()
MovieLens100K.raw_file_names
MovieLens100K.download()
MovieLens100K.read_raw()
Yelp2018
Yelp2018.url
Yelp2018.raw_file_names
Yelp2018.__init__()
Yelp2018.download()
Yelp2018.read_raw()
GeneralLoader
GeneralLoader.itermax
GeneralLoader.__init__()
GeneralLoader.itercnt
GeneralLoader.dataset
RandomNodeLoader
RandomNodeLoader.__init__()
RandomNodeLoader.get_node_indices()
NeighborLoader
NeighborLoader.__init__()
NeighborLoader.get_node_indices()
NeighborLoader.neighbor_sampling()
NeighborLoader.unique()
ClusterLoader
ClusterLoader.__init__()
ClusterLoader.partition()
ClusterLoader.get_node_indices()
DataLoader
DataLoader.__init__()
Advanced Features
ChunkManager
ChunkManager.__init__()
ChunkManager.metis_partition()
ChunkManager.partition_to_chunk()
NormalizeFeatures
Utilities & I/O
parse_txt_array()
read_txt_array()
read_planetoid_data()
read_npz()
read_graph()
read_heterograph()
read_node_label_hetero()
read_nodesplitidx_split_hetero()
coalesce()
degree()
contains_self_loops()
remove_self_loops()
segregate_self_loops()
add_self_loops()
add_remaining_self_loops()
contains_isolated_nodes()
remove_isolated_nodes()
get_laplacian()
is_jittor_sparse_tensor()
scatter()
induced_graph()
unique()
neighbor_sampler()
randomwalk_sampler()
one_hot()
maybe_num_nodes()
from_rdmol()
to_rdmol()
from_smiles()
to_smiles()
set_seed()
cheby()
random_splits()
drop_feature()
mask_edge()
preprocess_features()
compute_ppr()