-
Notifications
You must be signed in to change notification settings - Fork 0
/
lrBFGS.m
111 lines (94 loc) · 2.5 KB
/
lrBFGS.m
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
function [theta cost] = lrBFGS(x, y, option)
% Logistic Regression Solver: BFGS
% http://en.wikipedia.org/wiki/BFGS_method
% x -- input data, size = [m, n], m:samples number, n:feature dimension;
% y -- labels data, size = [m, 1], values=[-1 1], m:samples number;
% theta -- parameters, size = [n+1, 1], n:elements nubmer;
% cost -- cost
% option -- option struct
% max_itr: max iterators
% min_eps: min eps
% C: penalty factor
% debug: show debug message
% author -- amadeuzou AT gmail
% date -- 11/19/2013, Beijing, China
if nargin == 2
option.C = 1;
option.max_itr = 100;
option.min_eps = 1e-3;
option.debug = 1;
end
if ~isfield(option, 'C')
option.C = 1;
end
if ~isfield(option, 'max_itr')
option.max_itr = 100;
end
if ~isfield(option, 'min_eps')
option.min_eps = 1e-3;
end
if ~isfield(option, 'debug')
option.debug = 1;
end
[m, n] = size(x);
x = [ones(m, 1), x];
theta = zeros(n+1, 1);
theta_k = theta;
J = [];
H = eye(n+1);
I = eye(n+1);
lambda0 = 0;
step0 = 0.1;
itr = 0;
err = 0;
j = 0;
% gradient
[cost g] = lrCostFunc(x, y, theta, option.C);
% descent direction
d = -H*g;
while(1)
% linear search
param.x = x;
param.y = y;
param.theta = theta;
param.d = d;
param.C = option.C;
lamb = lrLinearSearch(@lrCostFuncLambda, param, lambda0, step0);
theta_k = theta + lamb.*d;
if j<n
j = j+1;
continue;
else
[cost gk] = lrCostFunc(x, y, theta_k, option.C);
p = theta_k - theta;
q = gk - g;
%H = H + (1+q'*H*q./(p'*q)).*(p*p'./(p'*q)) - (p*q'*H+H*q*p')./(p'*q);
ruo = 1/dot(p, q);%1/p'*q;
v = I - ruo*q*p';
H = v'*H*v + ruo*p*p';
g = gk;
d = -H*g;
j = 0;
itr = itr + 1;
err = norm(p(:));
theta = theta_k;
end
% cost record
J = [J; cost];
if(option.debug)
disp(['itr = ', num2str(itr), ', cost = ', num2str(cost), ', err = ', num2str(err)]);
end
if itr >= option.max_itr || err <= option.min_eps || norm(g)<=option.min_eps
break;
end
end
% draw cost cure
if(option.debug)
figure(1024)
plot(1:length(J), J, 'b-');
xlabel('iterators');
ylabel('cost');
end
function cost = lrCostFuncLambda(param, lambda)
theta = param.theta + lambda.*param.d;
cost = lrCostFunc(param.x, param.y, theta, param.C);