GaussianProcessSurrogate.h

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

#pragma once

#include "SurrogateModel.h"
#include "Standardizer.h"
#include "CovarianceInterface.h"
#include "GaussianProcess.h"

#include "LibtorchUtils.h"

class GaussianProcessSurrogate : public SurrogateModel, public CovarianceInterface
{
public:
  static InputParameters validParams();
  GaussianProcessSurrogate(const InputParameters & parameters);
  using SurrogateModel::evaluate;
  virtual Real evaluate(const std::vector<Real> & x) const;
  virtual void evaluate(const std::vector<Real> & x, std::vector<Real> & y) const;
  virtual Real evaluate(const std::vector<Real> & x, Real & std) const;
  virtual void
  evaluate(const std::vector<Real> & x, std::vector<Real> & y, std::vector<Real> & std) const;

  virtual void setupCovariance(UserObjectName _covar_name);

  StochasticTools::GaussianProcess & gp() { return _gp; }
  const StochasticTools::GaussianProcess & getGP() const { return _gp; }

private:
  StochasticTools::GaussianProcess & _gp;

  const torch::Tensor & _training_params;
};

#endif