InDet::TRTPIDNN Class Reference

#include <TRTPIDNN.h>

Collaboration diagram for InDet::TRTPIDNN:

Public Member Functions
	TRTPIDNN ()=default
virtual	~TRTPIDNN ()=default
const std::string &	getDefaultOutputNode () const
const std::string &	getDefaultOutputLabel () const
const std::map< std::string, std::map< std::string, double > > &	getScalarInputs () const
const std::map< std::string, std::map< std::string, std::vector< double > > > &	getVectorInputs () const
double	evaluate (std::map< std::string, std::map< std::string, double >> &scalarInputs, std::map< std::string, std::map< std::string, std::vector< double >>> &vectorInputs) const
double	evaluate (std::map< std::string, std::map< std::string, double >> &scalarInputs, std::map< std::string, std::map< std::string, std::vector< double >>> &vectorInputs, const std::string &outputNode, const std::string &outputLabel) const
StatusCode	configure (const std::string &json)

Private Attributes
std::unique_ptr< lwt::LightweightGraph >	m_nn
lwt::GraphConfig	m_nnConfig
std::map< std::string, std::map< std::string, double > >	m_scalarInputs
std::map< std::string, std::map< std::string, std::vector< double > > >	m_vectorInputs
std::string	m_outputNode
std::string	m_outputLabel

Detailed Description

Definition at line 28 of file TRTPIDNN.h.

Constructor & Destructor Documentation

TRTPIDNN()

InDet::TRTPIDNN::TRTPIDNN ( )

default

~TRTPIDNN()

virtual InDet::TRTPIDNN::~TRTPIDNN ( )

virtualdefault

Member Function Documentation

configure()

StatusCode InDet::TRTPIDNN::configure

(

const std::string &

json

)

Definition at line 35 of file TRTPIDNN.cxx.

                                                         {
  MsgStream log(Athena::getMessageSvc(),"TRTPIDNN");
  try {
    std::istringstream inputCfg(json);
    m_nnConfig = lwt::parse_json_graph(inputCfg);
  } catch (boost::property_tree::ptree_error& err) {
    log << MSG::ERROR << " NN not readable: " << err.what() << endmsg;
    return StatusCode::FAILURE;
  }
 
  try {
    m_nn = std::make_unique<lwt::LightweightGraph>(m_nnConfig);
  } catch (lwt::NNConfigurationException& err) {
    log << MSG::ERROR << " NN configuration failed: " << err.what() << endmsg;
    return StatusCode::FAILURE;
  }
 
  // set the default output node name
  if (m_nnConfig.outputs.empty() or m_nnConfig.outputs.begin()->second.labels.empty()) {
    log << MSG::ERROR << " unable to define NN output." << endmsg;
    return StatusCode::FAILURE;
  }
  m_outputNode = m_nnConfig.outputs.begin()->first;
  m_outputLabel = *(m_nnConfig.outputs[m_outputNode].labels.begin());
 
  // store templates of the structure of the inputs to the NN
  m_scalarInputs.clear();
  for (auto input : m_nnConfig.inputs) {
    m_scalarInputs[input.name] = {};
    for (const auto& variable : input.variables) {
      m_scalarInputs[input.name][variable.name] = input.defaults[variable.name];
    }
  }
  m_vectorInputs.clear();
  for (const auto& input : m_nnConfig.input_sequences) {
    m_vectorInputs[input.name] = {};
    for (const auto& variable : input.variables) {
      m_vectorInputs[input.name][variable.name] = {};
    }
  }
 
  return StatusCode::SUCCESS;
}

evaluate() [1/2]

double InDet::TRTPIDNN::evaluate ( std::map< std::string, std::map< std::string, double >> &  scalarInputs, std::map< std::string, std::map< std::string, std::vector< double >>> &  vectorInputs  ) const

inline

Definition at line 52 of file TRTPIDNN.h.

                                                                                          {
      return evaluate(scalarInputs, vectorInputs, m_outputNode, m_outputLabel);
    }

evaluate() [2/2]

double InDet::TRTPIDNN::evaluate	(	std::map< std::string, std::map< std::string, double >> &	scalarInputs,
		std::map< std::string, std::map< std::string, std::vector< double >>> &	vectorInputs,
		const std::string &	outputNode,
		const std::string &	outputLabel
	)		const

Definition at line 22 of file TRTPIDNN.cxx.

                                                                         {
  MsgStream log(Athena::getMessageSvc(),"TRTPIDNN");
  const auto result = m_nn->compute(scalarInputs, vectorInputs, outputNode);
  const auto itResult = result.find(outputLabel);
  if (itResult == result.end()) {
    log << MSG::ERROR << " unable to find output: node=" << outputNode << ", label=" << outputLabel << endmsg;
    return 0.5;
  }
  return itResult->second;
}

getDefaultOutputLabel()

const std::string& InDet::TRTPIDNN::getDefaultOutputLabel ( ) const

inline

Definition at line 37 of file TRTPIDNN.h.

                                                   {
      return m_outputLabel;
    }

getDefaultOutputNode()

const std::string& InDet::TRTPIDNN::getDefaultOutputNode ( ) const

inline

Definition at line 33 of file TRTPIDNN.h.

                                                  {
      return m_outputNode;
    }

getScalarInputs()

const std::map<std::string, std::map<std::string, double> >& InDet::TRTPIDNN::getScalarInputs ( ) const

inline

Definition at line 42 of file TRTPIDNN.h.

                                                                                {
      return m_scalarInputs;
    }

getVectorInputs()

const std::map<std::string, std::map<std::string, std::vector<double> > >& InDet::TRTPIDNN::getVectorInputs ( ) const

inline

Definition at line 47 of file TRTPIDNN.h.

                                                                                           {
      return m_vectorInputs;
    }

Member Data Documentation

m_nn

std::unique_ptr<lwt::LightweightGraph> InDet::TRTPIDNN::m_nn

private

Definition at line 66 of file TRTPIDNN.h.

m_nnConfig

lwt::GraphConfig InDet::TRTPIDNN::m_nnConfig

private

Definition at line 67 of file TRTPIDNN.h.

m_outputLabel

std::string InDet::TRTPIDNN::m_outputLabel

private

Definition at line 71 of file TRTPIDNN.h.

m_outputNode

std::string InDet::TRTPIDNN::m_outputNode

private

Definition at line 70 of file TRTPIDNN.h.

m_scalarInputs

std::map<std::string, std::map<std::string, double> > InDet::TRTPIDNN::m_scalarInputs

private

Definition at line 68 of file TRTPIDNN.h.

m_vectorInputs

std::map<std::string, std::map<std::string, std::vector<double> > > InDet::TRTPIDNN::m_vectorInputs

private

Definition at line 69 of file TRTPIDNN.h.

---

The documentation for this class was generated from the following files:

• TRTPIDNN.h
• TRTPIDNN.cxx