[db7908]: / Ensemble Learning / AdaBoost / buildStump.m

Download this file

17 lines (14 with data), 310 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
function stump = buildStump(X, y, weight)
D = size(X, 2); % Dimension
if nargin <= 2
weight = ones(size(X,1), 1);
end
cellDS = cell(D, 1);
Err = zeros(D, 1);
for i = 1:D
cellDS{i} = buildOneDStump(X(:,i), y, i, weight);
Err(i) = cellDS{i}.error;
end
[v, idx] = min(Err);
stump = cellDS{idx};
end