1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
/*
* linkFunction.hpp
* Header file for logitFunction and linearFunction class
*/
#include "seercommon.hpp"
class LinkFunction
{
public:
// Initialisation
LinkFunction(const arma::mat& _predictors, const arma::vec& _responses, const double _lambda = 0)
: predictors(_predictors), responses(_responses), lambda(_lambda)
{
}
// Likelihood and first derivative
double likelihood(const column_vector& b)
const;
column_vector gradient(const column_vector& b)
const;
protected:
arma::mat predictors;
arma::vec responses;
double lambda;
};
class LogitLikelihood : public LinkFunction
{
public:
LogitLikelihood(const arma::mat& _predictors, const arma::vec& _responses, const double _lambda = 0)
: LinkFunction(_predictors, _responses, _lambda)
{
}
double operator() (const column_vector& parameters_in) const;
};
class LogitLikelihoodGradient : public LinkFunction
{
public:
LogitLikelihoodGradient(const arma::mat& _predictors, const arma::vec& _responses, const double _lambda = 0)
: LinkFunction(_predictors, _responses, _lambda)
{
}
column_vector operator() (const column_vector& parameters_in) const;
};
class LinearLikelihood : public LinkFunction
{
public:
LinearLikelihood(const arma::mat& _predictors, const arma::vec& _responses, const double _lambda = 0)
: LinkFunction(_predictors, _responses, _lambda)
{
}
double operator() (const column_vector& parameters_in) const;
};
class LinearLikelihoodGradient : public LinkFunction
{
public:
LinearLikelihoodGradient(const arma::mat& _predictors, const arma::vec& _responses, const double _lambda = 0)
: LinkFunction(_predictors, _responses, _lambda)
{
}
column_vector operator() (const column_vector& parameters_in) const;
};
|