DRLRewardReporter.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 "DRLRewardReporter.h"
 
registerMooseObject("StochasticToolsApp", DRLRewardReporter);
 
InputParameters
DRLRewardReporter::validParams()
{
  InputParameters params = GeneralReporter::validParams();
  params += SurrogateModelInterface::validParams();
 
  params.addClassDescription("Reporter containing the reward values of a DRL controller trainer.");
  params.addRequiredParam<UserObjectName>(
      "drl_trainer_name", "The name of the RDL controller trainer which computes the rewards.");
 
  return params;
}
 
DRLRewardReporter::DRLRewardReporter(const InputParameters & parameters)
  : GeneralReporter(parameters),
    SurrogateModelInterface(this),
    _reward(declareValueByName<Real>("average_reward", REPORTER_MODE_ROOT)),
    _trainer(getSurrogateTrainer<LibtorchDRLControlTrainer>("drl_trainer_name"))
{
}
 
void
DRLRewardReporter::execute()
{
  _reward = _trainer.averageEpisodeReward();
}
 
#endif