Skip to main content
Back to top
Ctrl
+
K
Install
Installation
Tutorials
Quickstart
Graph Classification
Benchmarking mlx-graphs
Examples
Quantum-chemical property prediction with GNNs
API
Data
Data Objects
mlx_graphs.data.data.GraphData
mlx_graphs.data.data.HeteroGraphData
Batch Objects
mlx_graphs.data.batch.GraphDataBatch
mlx_graphs.data.batch.batch
mlx_graphs.data.batch.unbatch
mlx_graphs.data.collate.collate
Utils
mlx_graphs.data.utils.validate_list_of_graph_data
Datasets
mlx_graphs.datasets.Dataset
mlx_graphs.datasets.KarateClubDataset
mlx_graphs.datasets.PlanetoidDataset
mlx_graphs.datasets.QM7bDataset
mlx_graphs.datasets.TUDataset
mlx_graphs.datasets.SuperPixelDataset
mlx_graphs.datasets.OGBDataset
mlx_graphs.datasets.EllipticBitcoinDataset
mlx_graphs.datasets.MovieLens100K
mlx_graphs.datasets.IMDB
mlx_graphs.datasets.DBLP
Loaders
mlx_graphs.loaders.Dataloader
Transforms
mlx_graphs.transforms.BaseTransform
mlx_graphs.transforms.FeaturesNormalizedTransform
Neural Networks
mlx_graphs.nn.Linear
mlx_graphs.nn.MessagePassing
mlx_graphs.nn.GraphNetworkBlock
mlx_graphs.nn.SimpleConv
mlx_graphs.nn.GATConv
mlx_graphs.nn.GCNConv
mlx_graphs.nn.GINConv
mlx_graphs.nn.SAGEConv
mlx_graphs.nn.GeneralizedRelationalConv
mlx_graphs.nn.BatchNormalization
mlx_graphs.nn.InstanceNormalization
mlx_graphs.nn.LayerNormalization
mlx_graphs.nn.global_add_pool
mlx_graphs.nn.global_max_pool
mlx_graphs.nn.global_mean_pool
Utils
Transformations
mlx_graphs.utils.transformations.to_edge_index
mlx_graphs.utils.transformations.to_sparse_adjacency_matrix
mlx_graphs.utils.transformations.to_adjacency_matrix
mlx_graphs.utils.transformations.get_src_dst_features
mlx_graphs.utils.transformations.get_unique_edge_indices
mlx_graphs.utils.transformations.add_self_loops
mlx_graphs.utils.transformations.remove_self_loops
mlx_graphs.utils.transformations.to_undirected
mlx_graphs.utils.transformations.remove_duplicate_directed_edges
mlx_graphs.utils.transformations.coalesce
mlx_graphs.utils.transformations.get_isolated_nodes_mask
mlx_graphs.utils.transformations.has_isolated_nodes
mlx_graphs.utils.transformations.has_self_loops
Topology
mlx_graphs.utils.topology.is_undirected
mlx_graphs.utils.topology.is_directed
mlx_graphs.utils.topology.get_num_hops
Sorting
mlx_graphs.utils.sorting.sort_edge_index
mlx_graphs.utils.sorting.sort_edge_index_and_features
Validators
mlx_graphs.utils.validators.validate_adjacency_matrix
mlx_graphs.utils.validators.validate_edge_index
mlx_graphs.utils.validators.validate_edge_index_and_features
Scatter
mlx_graphs.utils.scatter.scatter
mlx_graphs.utils.scatter.scatter_add
mlx_graphs.utils.scatter.scatter_max
mlx_graphs.utils.scatter.scatter_mean
mlx_graphs.utils.scatter.scatter_softmax
mlx_graphs.utils.scatter.degree
mlx_graphs.utils.scatter.invert_sqrt_degree
Array ops
mlx_graphs.utils.array_ops.broadcast
mlx_graphs.utils.array_ops.expand
mlx_graphs.utils.array_ops.one_hot
mlx_graphs.utils.array_ops.pairwise_distances
mlx_graphs.utils.array_ops.index_to_mask
Convert
mlx_graphs.utils.convert.to_networkx
mlx_graphs.utils.convert.from_networkx
.rst
.pdf
Examples
Examples
#
Examples
Quantum-chemical property prediction with GNNs