Definition at line 69 of file TMVA_SOFIE_GNN_Parser.py.
◆ __init__()
| TMVA_SOFIE_GNN_Parser.MLPGraphIndependent.__init__ |
( |
| self, |
|
|
| name = "MLPGraphIndependent" ) |
◆ __call__()
| TMVA_SOFIE_GNN_Parser.MLPGraphIndependent.__call__ |
( |
| self, |
|
|
| inputs ) |
◆ _network
| TMVA_SOFIE_GNN_Parser.MLPGraphIndependent._network |
|
protected |
Initial value:= gn.modules.GraphIndependent(
edge_model_fn = lambda: snt.nets.MLP([LATENT_SIZE]*NUM_LAYERS, activate_final=True),
node_model_fn = lambda: snt.nets.MLP([LATENT_SIZE]*NUM_LAYERS, activate_final=True),
global_model_fn = lambda: snt.nets.MLP([LATENT_SIZE]*NUM_LAYERS, activate_final=True))
Definition at line 72 of file TMVA_SOFIE_GNN_Parser.py.