summaryrefslogtreecommitdiff
path: root/inst/lm/@stk_lm_cubic/feval.m
blob: 0925bae1aa4be7516b63107ab193bbd8cd31d209 (plain)
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
% FEVAL [overload base function]

% Copyright Notice
%
%    Copyright (C) 2014 SUPELEC
%
%    Author:  Julien Bect  <julien.bect@centralesupelec.fr>

% Copying Permission Statement
%
%    This file is part of
%
%            STK: a Small (Matlab/Octave) Toolbox for Kriging
%               (https://github.com/stk-kriging/stk/)
%
%    STK is free software: you can redistribute it and/or modify it under
%    the terms of the GNU General Public License as published by the Free
%    Software Foundation,  either version 3  of the License, or  (at your
%    option) any later version.
%
%    STK is distributed  in the hope that it will  be useful, but WITHOUT
%    ANY WARRANTY;  without even the implied  warranty of MERCHANTABILITY
%    or FITNESS  FOR A  PARTICULAR PURPOSE.  See  the GNU  General Public
%    License for more details.
%
%    You should  have received a copy  of the GNU  General Public License
%    along with STK.  If not, see <http://www.gnu.org/licenses/>.

function z = feval (lm, x) %#ok<INUSL>

x = double (x);

[n, d] = size (x);

% Use stk_lm_quadratic to compute the quadratic part
z2 = feval (stk_lm_quadratic, x);

% And now compute all monomials of degree exactly 3
z3 = zeros (n, d * (2 + d * (3 + d)) / 6);

c = 1;
for i = 1:d
    u = x(:, i);
    for j = i:d
        v = u .* x(:, j);
        for k = j:d
            z3(:, c) = v .* x(:, k);
            c = c + 1;
        end
    end
end

z = horzcat (z2, z3);

end % function


%!test
%! n = 15; d = 4;
%! x = stk_sampling_randunif (n, d);
%! P = feval (stk_lm_cubic (), x);
%! assert (isequal (size (P), [n, 1 + d * (11 + d * (6 + d)) / 6]))