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