negbin.hpp

Go to the documentation of this file.

// toolboxes/stats/include/nk/stats/distributions/negbin.hpp
//
// Negative binomial (number of failures before the r-th success).
 
#pragma once
 
#include <memory_resource>
#include <nk/value/value.hpp>
 
#include <tuple>
 
namespace nk {

 namespace ops { class RngContext; } }
 
namespace nk::stats {

Value nbinpdf(const Value &k, double r, double p,
              std::pmr::memory_resource *mr = nullptr);

Value nbincdf(const Value &k, double r, double p,
              std::pmr::memory_resource *mr = nullptr);

Value nbininv(const Value &q, double r, double p,
              std::pmr::memory_resource *mr = nullptr);

Value nbinrnd(::nk::ops::RngContext &rng, double r, double p, size_t rows = 1, size_t cols = 1,
              std::pmr::memory_resource *mr = nullptr);

std::tuple<double, double> nbinstat(double r, double p);

Value nbinfit(const Value &x, std::pmr::memory_resource *mr = nullptr);

Value nbinfit_ci(const Value &x, double alpha = 0.05,
                 std::pmr::memory_resource *mr = nullptr);
 

} // namespace nk::stats