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
|
// Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
// Copyright (C) 2014 - Scilab Enterprises - Paul Bignier
//
// This file must be used under the terms of the CeCILL.
// This source file is licensed as described in the file COPYING, which
// you should have received as part of this distribution. The terms
// are also available at
// http://www.cecill.info/licences/Licence_CeCILL_V2.1-en.txt
function [a, b] = nanreglin(x, y, dflag)
// Solves a linear regression
// y=a(p,q)*x+b(p,1) + epsilon where y or x contain NaNs.
// x : matrix (q,n) and y matrix (p,n)
// dflag is optional if 1 a display of the result is done
//!
[lhs, rhs] = argn(0);
if rhs < 2 then
error(msprintf(_("%s: Wrong number of input arguments: %d to %d expected.\n"),"nanreglin",2,3))
end
if rhs <= 2 then
dflag = 0;
end
[n1, n2] = size(x);
[p1, p2] = size(y);
if n2 <> p2 then
error(msprintf(_("%s: Incompatible input arguments #%d and #%d: Same column dimensions expected.\n"),"nanreglin",1,2));
end
if ~(or(isnan(x)) | or(isnan(y))) then
error(msprintf(_("%s: No NaNs detected, please use %s() instead.\n"), "nanreglin", "reglin"))
end
a = zeros(p1, n1);
b = zeros(p1, 1);
for i=1:p1
// A column of x defines an element of y, but each line of y defines an independent problem.
// If x2(:, j) or y2(i, j) contains a %nan, then both x2(:, j) and y2(j) are removed.
y2 = y(i, find(~isnan(y(i,:))));
x2 = x(:, find(~isnan(y(i,:))));
nanX = isnan(x);
if or(isnan(x)) then // At least one NaN is x or y.
columns = floor((find(nanX==%t)-1)./(n1+1)+1);
x2(:, columns) = [];
y2(1, columns) = [];
end
[a(i, :), b(i)] = reglin(x2, y2, dflag);
end
endfunction
|
