Public Member Functions | Private Attributes | Static Private Attributes | List of all members
LSTME::Model Class Reference

#include "/cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N20-11-28/TensorFlowEvaluator/LSTME/model/Model.h"

Public Member Functions

 Model (const std::string &savedir)
 
LSTMEnergy predict (const VarDict &varDict)
 

Private Attributes

SimpleModel simpleModel
 

Static Private Attributes

static const std::vector< InputConfigKeyssliceInputs
 
static const std::vector< InputConfigKeysprongInputs
 
static const std::vector< std::stringoutputs
 

Detailed Description

Definition at line 15 of file Model.h.

Constructor & Destructor Documentation

LSTME::Model::Model ( const std::string savedir)

Definition at line 23 of file Model.cxx.

25 { }
static const std::vector< std::string > outputs
Definition: Model.h:22
SimpleModel simpleModel
Definition: Model.h:18
static const std::vector< InputConfigKeys > sliceInputs
Definition: Model.h:20
static const std::vector< InputConfigKeys > prongInputs
Definition: Model.h:21

Member Function Documentation

LSTMEnergy LSTME::Model::predict ( const VarDict varDict)

Definition at line 27 of file Model.cxx.

References check_time_usage::float, outputs, SimpleModel::predict(), and simpleModel.

Referenced by LSTME::FillLSTME::predictEnergy().

28 {
29  std::vector<tensorflow::Tensor> outputs = simpleModel.predict(varDict);
30 
31  const float primaryE = outputs[0].tensor<float,2>()(0, 0);
32  const float totalE = outputs[1].tensor<float,2>()(0, 0);
33 
34  return LSTMEnergy{ primaryE, totalE };
35 }
static const std::vector< std::string > outputs
Definition: Model.h:22
SimpleModel simpleModel
Definition: Model.h:18
std::vector< tensorflow::Tensor > predict(const VarDict &varDict) const

Member Data Documentation

const std::vector< std::string > LSTME::Model::outputs
staticprivate

Definition at line 22 of file Model.h.

Referenced by predict().

const std::vector< InputConfigKeys > LSTME::Model::prongInputs
staticprivate

Definition at line 21 of file Model.h.

SimpleModel LSTME::Model::simpleModel
private

Definition at line 18 of file Model.h.

Referenced by predict().

const std::vector< InputConfigKeys > LSTME::Model::sliceInputs
staticprivate

Definition at line 20 of file Model.h.


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