topobench.nn package

Empty file to make the directory a package.

Subpackages

• topobench.nn.backbones package
  • CCCN
    • CCCN.__init__()
    • CCCN.forward()
  • CW
    • CW.__init__()
    • CW.forward()
  • EDGNN
    • EDGNN.__init__()
    • EDGNN.forward()
    • EDGNN.reset_parameters()
  • EquivSetConv
    • EquivSetConv.__init__()
    • EquivSetConv.forward()
    • EquivSetConv.reset_parameters()
  • GPSEncoder
    • GPSEncoder.__init__()
    • GPSEncoder.forward()
  • GraphMLP
    • GraphMLP.__init__()
    • GraphMLP.forward()
  • HOPSE
    • HOPSE.__init__()
    • HOPSE.forward()
  • HOPSELayer
    • HOPSELayer.__init__()
    • HOPSELayer.forward()
    • HOPSELayer.update()
  • IdentityGAT
    • IdentityGAT.__init__()
    • IdentityGAT.forward()
  • IdentityGCN
    • IdentityGCN.__init__()
    • IdentityGCN.forward()
  • IdentityGIN
    • IdentityGIN.__init__()
    • IdentityGIN.forward()
  • IdentitySAGE
    • IdentitySAGE.__init__()
    • IdentitySAGE.forward()
  • JumpLinkConv
    • JumpLinkConv.__init__()
    • JumpLinkConv.forward()
    • JumpLinkConv.reset_parameters()
  • MLP
    • MLP.__init__()
    • MLP.build_mlp_layers()
    • MLP.build_norm_layers()
    • MLP.forward()
  • MeanDegConv
    • MeanDegConv.__init__()
    • MeanDegConv.forward()
    • MeanDegConv.reset_parameters()
  • Mlp
    • Mlp.__init__()
    • Mlp.forward()
  • NSDEncoder
    • NSDEncoder.__init__()
    • NSDEncoder.forward()
    • NSDEncoder.get_sheaf_model()
  • PlainMLP
    • PlainMLP.__init__()
    • PlainMLP.forward()
    • PlainMLP.reset_parameters()
  • RedrawProjection
    • RedrawProjection.__init__()
    • RedrawProjection.redraw_projections()
  • SCCNNCustom
    • SCCNNCustom.__init__()
    • SCCNNCustom.forward()
  • SCCNNLayer
    • SCCNNLayer.__init__()
    • SCCNNLayer.aggr_norm_func()
    • SCCNNLayer.chebyshev_conv()
    • SCCNNLayer.forward()
    • SCCNNLayer.reset_parameters()
    • SCCNNLayer.update()
  • TopoTune
    • TopoTune.__init__()
    • TopoTune.aggregate_inter_nbhd()
    • TopoTune.forward()
    • TopoTune.generate_membership_vectors()
    • TopoTune.get_nbhd_cache()
    • TopoTune.interrank_expand()
    • TopoTune.interrank_gnn_forward()
    • TopoTune.intrarank_expand()
    • TopoTune.intrarank_gnn_forward()
  • TopoTune_OneHasse
    • TopoTune_OneHasse.__init__()
    • TopoTune_OneHasse.aggregate_inter_nbhd()
    • TopoTune_OneHasse.all_nbhds_expand()
    • TopoTune_OneHasse.all_nbhds_gnn_forward()
    • TopoTune_OneHasse.forward()
    • TopoTune_OneHasse.generate_membership_vectors()
  • customMLP
    • customMLP.__init__()
    • customMLP.flops()
    • customMLP.forward()
    • customMLP.reset_parameters()
  • Subpackages
    • topobench.nn.backbones.cell package
      • CCCN
      • CW
      • Submodules
    • topobench.nn.backbones.combinatorial package
      • HOPSE
      • HOPSELayer
      • TopoTune
      • TopoTune_OneHasse
      • Submodules
    • topobench.nn.backbones.graph package
      • GPSEncoder
      • GraphMLP
      • IdentityGAT
      • IdentityGCN
      • IdentityGIN
      • IdentitySAGE
      • Mlp
      • NSDEncoder
      • RedrawProjection
      • Subpackages
      • Submodules
    • topobench.nn.backbones.hypergraph package
      • EDGNN
      • EquivSetConv
      • JumpLinkConv
      • MeanDegConv
      • PlainMLP
      • customMLP
      • Submodules
    • topobench.nn.backbones.non_relational package
      • MLP
      • Submodules
    • topobench.nn.backbones.simplicial package
      • SCCNNCustom
      • SCCNNLayer
      • Submodules
• topobench.nn.encoders package
  • AllCellFeatureEncoder
    • AllCellFeatureEncoder.__init__()
    • AllCellFeatureEncoder.forward()
  • DGMStructureFeatureEncoder
    • DGMStructureFeatureEncoder.__init__()
    • DGMStructureFeatureEncoder.forward()
  • FlatEncoder
    • FlatEncoder.__init__()
    • FlatEncoder.forward()
  • HOPSEFeatureEncoder
    • HOPSEFeatureEncoder.__init__()
    • HOPSEFeatureEncoder.forward()
  • Submodules
    • topobench.nn.encoders.all_cell_encoder module
      • AbstractFeatureEncoder
      • AllCellFeatureEncoder
      • BaseEncoder
      • GraphNorm
    • topobench.nn.encoders.base module
      • AbstractFeatureEncoder
      • abstractmethod()
    • topobench.nn.encoders.dgm_encoder module
      • AbstractFeatureEncoder
      • BaseEncoder
      • DGMStructureFeatureEncoder
      • DGM_d
    • topobench.nn.encoders.flat_encoder module
      • AbstractFeatureEncoder
      • FlatEncoder
    • topobench.nn.encoders.hopse_encoder module
      • AbstractFeatureEncoder
      • AtomEncoder
      • BondEncoder
      • HOPSEFeatureEncoder
      • MLP
      • SimpleAtomEncoder
      • SimpleBondEncoder
    • topobench.nn.encoders.kdgm module
      • DGM_d
      • pairwise_euclidean_distances()
      • pairwise_poincare_distances()
• topobench.nn.readouts package
  • AbstractZeroCellReadOut
    • AbstractZeroCellReadOut.__init__()
    • AbstractZeroCellReadOut.compute_logits()
    • AbstractZeroCellReadOut.forward()
  • HOPSEReadout
    • HOPSEReadout.__init__()
    • HOPSEReadout.compute_logits()
    • HOPSEReadout.forward()
  • MLPReadout
    • MLPReadout.__init__()
    • MLPReadout.forward()
  • NoReadOut
    • NoReadOut.__init__()
    • NoReadOut.forward()
  • PropagateSignalDown
    • PropagateSignalDown.__init__()
    • PropagateSignalDown.forward()
  • Submodules
    • topobench.nn.readouts.base module
      • AbstractZeroCellReadOut
      • abstractmethod()
      • scatter()
    • topobench.nn.readouts.hopse module
      • AbstractZeroCellReadOut
      • HOPSEReadout
      • scatter()
    • topobench.nn.readouts.identical module
      • AbstractZeroCellReadOut
      • NoReadOut
    • topobench.nn.readouts.mlp_readout module
      • MLPBackbone
      • MLPReadout
      • scatter()
    • topobench.nn.readouts.propagate_signal_down module
      • AbstractZeroCellReadOut
      • PropagateSignalDown
• topobench.nn.wrappers package
  • AbstractWrapper
    • AbstractWrapper.__init__()
    • AbstractWrapper.forward()
    • AbstractWrapper.residual_connection()
  • CANWrapper
    • CANWrapper.forward()
  • CCCNWrapper
    • CCCNWrapper.forward()
  • CCXNWrapper
    • CCXNWrapper.forward()
  • CWNWrapper
    • CWNWrapper.forward()
  • GNNWrapper
    • GNNWrapper.forward()
  • GraphMLPWrapper
    • GraphMLPWrapper.forward()
  • HOPSEWrapper
    • HOPSEWrapper.__init__()
    • HOPSEWrapper.forward()
  • HypergraphWrapper
    • HypergraphWrapper.forward()
  • PointcloudWrapper
    • PointcloudWrapper.forward()
  • SANWrapper
    • SANWrapper.forward()
  • SCCNNWrapper
    • SCCNNWrapper.forward()
  • SCCNWrapper
    • SCCNWrapper.forward()
  • SCNWrapper
    • SCNWrapper.forward()
    • SCNWrapper.normalize_matrix()
  • TuneWrapper
    • TuneWrapper.forward()
  • Subpackages
    • topobench.nn.wrappers.cell package
      • CANWrapper
      • CCCNWrapper
      • CCXNWrapper
      • CWNWrapper
      • Submodules
    • topobench.nn.wrappers.combinatorial package
      • HOPSEWrapper
      • TuneWrapper
      • Submodules
    • topobench.nn.wrappers.graph package
      • GNNWrapper
      • GraphMLPWrapper
      • Submodules
    • topobench.nn.wrappers.hypergraph package
      • HypergraphWrapper
      • Submodules
    • topobench.nn.wrappers.pointcloud package
      • PointcloudWrapper
      • Submodules
    • topobench.nn.wrappers.simplicial package
      • SANWrapper
      • SCCNNWrapper
      • SCCNWrapper
      • SCNWrapper
      • Submodules
  • Submodules
    • topobench.nn.wrappers.base module
      • ABC
      • AbstractWrapper
      • abstractmethod()