topobench.nn.wrappers.hypergraph package

Submodules

topobench.nn.wrappers.hypergraph.hypergraph_wrapper module

Wrapper for the hypergraph models.

class topobench.nn.wrappers.hypergraph.hypergraph_wrapper.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.

Module contents

Wrappers for hypergrpah neural networks with automated exports.

class topobench.nn.wrappers.hypergraph.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.