LibtorchDRLLogProbabilityPostprocessor.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#ifdef MOOSE_LIBTORCH_ENABLED
 
#include "LibtorchDRLLogProbabilityPostprocessor.h"
 
registerMooseObject("StochasticToolsApp", LibtorchDRLLogProbabilityPostprocessor);
 
InputParameters
LibtorchDRLLogProbabilityPostprocessor::validParams()
{
  InputParameters params = GeneralPostprocessor::validParams();
 
  params.addClassDescription(
      "Computes the logarithmic probability of the action in a given LibtorchDRLController.");
 
  params.addRequiredParam<std::string>("control_name",
                                       "The name of the LibtorchNeuralNetControl object.");
  params.addParam<unsigned int>("signal_index",
                                0,
                                "The index of the signal from the LibtorchNeuralNetControl object. "
                                "This assumes indexing between [0,num_signals).");
  return params;
}
 
LibtorchDRLLogProbabilityPostprocessor::LibtorchDRLLogProbabilityPostprocessor(
    const InputParameters & params)
  : GeneralPostprocessor(params), _signal_index(getParam<unsigned int>("signal_index"))
 
{
}
 
void
LibtorchDRLLogProbabilityPostprocessor::initialSetup()
{
  GeneralPostprocessor::initialSetup();
 
  _libtorch_nn_control =
      dynamic_cast<LibtorchDRLControl *>(_fe_problem.getControlWarehouse()
                                             .getActiveObject(getParam<std::string>("control_name"))
                                             .get());
  if (!_libtorch_nn_control)
    paramError("control_name",
               "The supplied control object is not derived from LibtorchDRLControl!");
 
  if (_libtorch_nn_control->numberOfControlSignals() <= _signal_index)
    paramError("signal_index",
               "The given control object only has ",
               _libtorch_nn_control->numberOfControlSignals(),
               " control signals!");
}
 
Real
LibtorchDRLLogProbabilityPostprocessor::getValue() const
{
  // Return the value of the control signal
  return _libtorch_nn_control->getSignalLogProbability(_signal_index);
}
 
#endif