VLN::VLNEnergyModel Class Reference

#include <VLNEnergyModel.h>

Public Member Functions
	VLNEnergyModel (const std::string &savedir)
VLNEnergy	predict (const VarDict &varDict) const

Private Attributes
TFModel	model

Static Private Attributes
static const std::vector< InputConfigKeys >	scalarInputKeys
static const std::vector< InputConfigKeys >	vectorInputKeys
static const std::vector< std::string >	outputKeys

Detailed Description

Definition at line 13 of file VLNEnergyModel.h.

Constructor & Destructor Documentation

VLN::VLNEnergyModel::VLNEnergyModel ( const std::string &  savedir )

explicit

Definition at line 21 of file VLNEnergyModel.cxx.

    : model(savedir, scalarInputKeys, vectorInputKeys, outputKeys)
{ }

Member Function Documentation

VLNEnergy VLN::VLNEnergyModel::predict

(

const VarDict &

varDict

)

const

Definition at line 25 of file VLNEnergyModel.cxx.

{
    std::vector<tensorflow::Tensor> outputs = model.predict(vars);

    const float primaryE = outputs[0].tensor<float,2>()(0, 0);
    const float totalE   = outputs[1].tensor<float,2>()(0, 0);

    return VLNEnergy{ primaryE, totalE };
}

Member Data Documentation

TFModel VLN::VLNEnergyModel::model

private

Definition at line 16 of file VLNEnergyModel.h.

const std::vector< std::string > VLN::VLNEnergyModel::outputKeys

staticprivate

Definition at line 20 of file VLNEnergyModel.h.

const std::vector< InputConfigKeys > VLN::VLNEnergyModel::scalarInputKeys

staticprivate

Definition at line 18 of file VLNEnergyModel.h.

const std::vector< InputConfigKeys > VLN::VLNEnergyModel::vectorInputKeys

staticprivate

Definition at line 19 of file VLNEnergyModel.h.

---

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

• dunereco/dunereco/VLNets/models/zoo/VLNEnergyModel.h
• dunereco/dunereco/VLNets/models/zoo/VLNEnergyModel.cxx