laplace_latent_neg_binomial_2_log_rng.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_MIX_PROB_LAPLACE_LATENT_NEG_BINOMIAL_2_LOG_RNG_HPP
#define STAN_MATH_MIX_PROB_LAPLACE_LATENT_NEG_BINOMIAL_2_LOG_RNG_HPP
 
#include <stan/math/mix/functor/laplace_base_rng.hpp>
#include <stan/math/mix/functor/laplace_likelihood.hpp>
#include <stan/math/mix/prob/laplace_marginal_neg_binomial_2_log_lpmf.hpp>
 
namespace stan {
namespace math {
 
template <typename Eta, typename ThetaVec, typename Mean, typename CovarFun,
          typename CovarArgs, typename RNG,
          require_eigen_vector_t<ThetaVec>* = nullptr>
inline Eigen::VectorXd laplace_latent_tol_neg_binomial_2_log_rng(
    const std::vector<int>& y, const std::vector<int>& y_index, Eta&& eta,
    Mean&& mean, CovarFun&& covariance_function, CovarArgs&& covar_args,
    ThetaVec&& theta_0, const double tolerance, const int max_num_steps,
    const int hessian_block_size, const int solver,
    const int max_steps_line_search, RNG& rng, std::ostream* msgs) {
  laplace_options_user_supplied ops{hessian_block_size,    solver,
                                    max_steps_line_search, tolerance,
                                    max_num_steps,         value_of(theta_0)};
  return laplace_base_rng(
      neg_binomial_2_log_likelihood{},
      std::forward_as_tuple(std::forward<Eta>(eta), y, y_index,
                            std::forward<Mean>(mean)),
      std::forward<CovarFun>(covariance_function),
      std::forward<CovarArgs>(covar_args), ops, rng, msgs);
}
 
template <typename Eta, typename Mean, typename CovarFun, typename CovarArgs,
          typename RNG>
inline Eigen::VectorXd laplace_latent_neg_binomial_2_log_rng(
    const std::vector<int>& y, const std::vector<int>& y_index, Eta&& eta,
    Mean&& mean, CovarFun&& covariance_function, CovarArgs&& covar_args,
    RNG& rng, std::ostream* msgs) {
  return laplace_base_rng(
      neg_binomial_2_log_likelihood{},
      std::forward_as_tuple(std::forward<Eta>(eta), y, y_index,
                            std::forward<Mean>(mean)),
      std::forward<CovarFun>(covariance_function),
      std::forward<CovarArgs>(covar_args), laplace_options_default{}, rng,
      msgs);
}
 
}  // namespace math
}  // namespace stan
 
#endif