UpperConfidenceBound.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* 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

#include "UpperConfidenceBound.h"
#include <cmath>

registerMooseObject("StochasticToolsApp", UpperConfidenceBound);

InputParameters
UpperConfidenceBound::validParams()
{
  InputParameters params = ParallelAcquisitionFunctionBase::validParams();
  params.addClassDescription("Upper Confidence Bound acquisition function.");
  params.addRangeCheckedParam<Real>(
      "tuning", 1.0, "tuning > 0", "Tuning parameter to control exploration vs exploitation.");
  return params;
}

UpperConfidenceBound::UpperConfidenceBound(const InputParameters & parameters)
  : ParallelAcquisitionFunctionBase(parameters), _tuning(getParam<Real>("tuning"))
{
}

void
UpperConfidenceBound::computeAcquisitionInternal(
    std::vector<Real> & acq,
    const std::vector<Real> & gp_mean,
    const std::vector<Real> & gp_std,
    const std::vector<std::vector<Real>> & /*test_inputs*/,
    const std::vector<std::vector<Real>> & /*train_inputs*/,
    const std::vector<Real> & /*generic*/) const
{
  for (unsigned int i = 0; i < gp_mean.size(); ++i)
    acq[i] = gp_mean[i] + _tuning * gp_std[i];
}