|
ATLAS Offline Software
|
Go to the documentation of this file.
5 #ifndef PhotonVertexSelection_PhotonVertexSelectionTool_h
6 #define PhotonVertexSelection_PhotonVertexSelectionTool_h
22 #include <onnxruntime_cxx_api.h>
63 std::unique_ptr<TMVA::Reader>
m_mva1;
64 std::unique_ptr<TMVA::Reader>
m_mva2;
89 std::tuple<std::shared_ptr<Ort::Session>, Ort::AllocatorWithDefaultOptions>
setONNXSession(Ort::Env&
env, std::string modelFilePath);
91 std::tuple<std::vector<int64_t>, std::vector<const char*>>
getInputNodes(
const std::shared_ptr<Ort::Session> sessionHandle, Ort::AllocatorWithDefaultOptions& allocator);
93 std::tuple<std::vector<int64_t>, std::vector<const char*>>
getOutputNodes(
const std::shared_ptr<Ort::Session> sessionHandle, Ort::AllocatorWithDefaultOptions& allocator);
95 float getScore(
int nVars, std::vector<std::vector<float>> input_data,
const std::shared_ptr<Ort::Session> sessionHandle, std::vector<int64_t> input_node_dims, std::vector<const char*> input_node_names, std::vector<const char*> output_node_names)
const;
103 static bool sortMLP(
const std::pair<const xAOD::Vertex*, float> &
a,
const std::pair<const xAOD::Vertex*, float> &
b);
148 #endif // PhotonVertexSelection_PhotonVertexSelectionTool_h
Select isolated Photons, Electrons and Muons.
::StatusCode StatusCode
StatusCode definition for legacy code.
Class describing a Vertex.