nb.hpp
Go to the documentation of this file.
1 
10 #pragma once
11 #include <vector>
12 #include "utils.hpp"
13 
14 namespace target {
15 
16  using raggedArray = std::vector<arma::vec>;
17 
18 
19  raggedArray pcond(const arma::uvec &idx,
20  const arma::mat &x,
21  const arma::uvec &xlev,
22  const arma::vec &weights,
23  double laplacesmooth);
24 
25  std::vector<raggedArray> nb(arma::vec y,
26  arma::mat x,
27  arma::uvec xlev = arma::uvec(),
28  arma::vec ylev = arma::vec(),
29  arma::vec weights = arma::vec(),
30  double laplacesmooth = 1.0);
31 
32  arma::mat prednb(const arma::mat &X,
33  const raggedArray &condprob,
34  raggedArray xord,
35  arma::uvec multinomial,
36  arma::vec prior = arma::vec(),
37  double threshold = 1E-3);
38 
39 } // namespace target
Various utility functions and constants.