![]() |
ATLAS Offline Software
|
The NodeConnector is indicating whether two space points inside a bucket, the graph nodes, shall have a connection in their graph neural net representation. More...
#include <NodeConnector.h>
Public Types | |
using | Bucket_t = NodeFeature::Bucket_t |
using | Evaluator_t = std::function< bool(const Bucket_t &, size_t, size_t)> |
Function type to connect two space points in a bucket. More... | |
Public Member Functions | |
NodeConnector (const std::string &cName, const Evaluator_t conFunc) | |
Standard constructor taking the name of the node connector & a connector function definition. More... | |
const std::string & | name () const |
Returns the name of the node connector. More... | |
bool | connect (const Bucket_t &bucket, size_t i, size_t j) const |
returns the decision of the connector function More... | |
Private Attributes | |
std::string | m_name {} |
Evaluator_t | m_func {[](const Bucket_t, size_t, size_t) { return false; }} |
The NodeConnector is indicating whether two space points inside a bucket, the graph nodes, shall have a connection in their graph neural net representation.
In short terms, the node connector is std::function with the bucket & the two indices to the space point inside as input arguments and then returning a boolean decision whether the connection shall be built. In order, to verify that two instances of the neural net graph are identical, the node connector also has a name as attribute.
Definition at line 19 of file NodeConnector.h.
Definition at line 21 of file NodeConnector.h.
using MuonML::NodeConnector::Evaluator_t = std::function<bool(const Bucket_t&, size_t, size_t)> |
Function type to connect two space points in a bucket.
The signature takes the reference to the bucket and then the two indices of the space points which shall be connected. The function needs to return true or false
Definition at line 25 of file NodeConnector.h.
|
inline |
Standard constructor taking the name of the node connector & a connector function definition.
cName: Name of the connector function
conFunc: Definition of the connector function
Definition at line 30 of file NodeConnector.h.
returns the decision of the connector function
bucket | Reference to the space point bucket |
i | Index of the first space point to connect |
j | Index of the second space point to connect |
Definition at line 40 of file NodeConnector.h.
|
inline |
Returns the name of the node connector.
Definition at line 33 of file NodeConnector.h.
|
private |
Definition at line 46 of file NodeConnector.h.
|
private |
Definition at line 45 of file NodeConnector.h.