Graph Wavelet Encoder

Graph wavelet scattering transform encoder for PyTorch Geometric graphs.

Computes multi-scale wavelet features from graph structure and node signals using lazy random-walk diffusion.

Installation

pip install graph-wavelet-encoder

For development (editable install):

pip install -e .

With dev dependencies (pytest, etc.):

pip install -e ".[dev]"

Usage

importtorchfromtorch_geometric.dataimportData, Batchfromgraph_wavelet_encoderimportGraphWaveletEncoder# Single graph or batched PyG Data# Expects: .x (node features), .edge_index, .batch (for batched graphs)encoder=GraphWaveletEncoder(
scales=(1, 2, 4, 8, 16),
sigma=2.0,
device=torch.device("cuda"iftorch.cuda.is_available() else"cpu"),
)
features=encoder.encode(graph) # [batch_size, num_nodes, num_features_per_node]

The encoder uses a lazy random-walk matrix and produces zeroth-, first-, and second-order scattering coefficients. See the docstrings in graph_wavelet_encoder.encoder for details.

Testing and benchmarking

pytest tests/test_encoder.py -v -s

License

Yale Licence