topobench.nn.wrappers.combinatorial package

Wrappers for combinatorial neural networks with automated exports.

class TuneWrapper(backbone, **kwargs)

Bases: AbstractWrapper

Wrapper for the TopoTune model.

This wrapper defines the forward pass of the TopoTune model.

forward(batch)

Forward pass for the Tune wrapper.

Parameters:

batchtorch_geometric.data.Data

Batch object containing the batched data.

Returns:

dict

Dictionary containing the updated model output.

Submodules

• topobench.nn.wrappers.combinatorial.tune_wrapper module
  • AbstractWrapper
    • AbstractWrapper.__init__()
    • AbstractWrapper.forward()
    • AbstractWrapper.residual_connection()
  • TuneWrapper
    • TuneWrapper.forward()