mlreco.models.layers.gnn.factories module¶
-
mlreco.models.layers.gnn.factories.gnn_model_construct(cfg, model_name='gnn_model', **kwargs)[source]¶ Instanties the appropriate GNN message passing model from the provided configuration.
- Parameters
dict – ‘name’: <Name of the model> <other entries that specify the model properties>
- Returns
Instantiated model
- Return type
object
-
mlreco.models.layers.gnn.factories.node_encoder_construct(cfg, model_name='node_encoder', **kwargs)[source]¶ Instanties the appropriate node encoder from the provided configuration.
- Parameters
dict – ‘name’: <Name of the node encoder> <other entries that specify the encoder properties>
- Returns
Instantiated node encoder
- Return type
object
-
mlreco.models.layers.gnn.factories.edge_encoder_construct(cfg, model_name='edge_encoder', **kwargs)[source]¶ Instanties the appropriate edge encoder from the provided configuration.
- Parameters
dict – ‘name’: <Name of the edge encoder> <other entries that specify the encoder properties>
- Returns
Instantiated edge encoder
- Return type
object
-
mlreco.models.layers.gnn.factories.node_loss_construct(cfg, model_name='node_loss', **kwargs)[source]¶ Instanties the appropriate node loss from the provided configuration.
- Parameters
dict – ‘name’: <Name of the node loss function> <other entries that specify the encoder properties>
- Returns
Instantiated node loss
- Return type
object
-
mlreco.models.layers.gnn.factories.edge_loss_construct(cfg, model_name='edge_loss', **kwargs)[source]¶ Instanties the appropriate edge loss from the provided configuration.
- Parameters
dict – ‘name’: <Name of the edge loss function> <other entries that specify the encoder properties>
- Returns
Instantiated edge loss
- Return type
object
-
mlreco.models.layers.gnn.factories.gnn_model_dict()[source]¶ Imports and returns dictionary of valid GNN message passing models.
Args: :returns: Dictionary of valid GNN message passing models :rtype: dict
-
mlreco.models.layers.gnn.factories.node_encoder_dict()[source]¶ Imports and returns dictionary of valid node encoders.
Args: :returns: Dictionary of valid node encoders :rtype: dict
-
mlreco.models.layers.gnn.factories.edge_encoder_dict()[source]¶ Imports and returns dictionary of valid edge encoders.
Args: :returns: Dictionary of valid edge encoders :rtype: dict