Add prediction function for one-vs-all classification
This commit is contained in:
parent
cf0d25440c
commit
3bf3d9fdc3
1 changed files with 10 additions and 13 deletions
|
@ -30,11 +30,8 @@ X = [ones(m, 1) X];
|
||||||
% for each row.
|
% for each row.
|
||||||
%
|
%
|
||||||
|
|
||||||
|
[~, p] = max(X * all_theta', [], 2);
|
||||||
|
%disp(size(p));
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
% =========================================================================
|
% =========================================================================
|
||||||
|
|
||||||
|
|
Reference in a new issue