topobench.nn.wrappers.hypergraph package

Wrappers for hypergrpah neural networks with automated exports.

class HypergraphWrapper(backbone, **kwargs)

Bases: AbstractWrapper

Wrapper for the hypergraph models.

This wrapper defines the forward pass of the model. The hypergraph model return the embeddings of the cells of rank 0, and 1 (the hyperedges).

forward(batch)

Forward pass for the hypergraph wrapper.

Parameters:

batchtorch_geometric.data.Data

Batch object containing the batched data.

Returns:

dict

Dictionary containing the updated model output.

Submodules

• topobench.nn.wrappers.hypergraph.hypergraph_wrapper module
  • AbstractWrapper
    • AbstractWrapper.__init__()
    • AbstractWrapper.forward()
    • AbstractWrapper.residual_connection()
  • HypergraphWrapper
    • HypergraphWrapper.forward()