comparison core/magnatagatune/mlr_unittest.m @ 0:e9a9cd732c1e tip

first hg version after svn
author wolffd
date Tue, 10 Feb 2015 15:05:51 +0000
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:e9a9cd732c1e
1 function mlr_unittest(X, Yrel)
2
3 % Loss values to test
4 LOSS = {'AUC', 'Prec@k', 'MAP', 'MRR', 'NDCG'};
5
6
7 % Regularization values to test
8 REG = [0,1,2,3];
9
10 % Batch sizes to test
11 BATCH = [0 1 5];
12
13 % Diagonal settings
14 DIAG = [0 1];
15
16 figure(1);
17 for l = 1:length(LOSS)
18 display(['Testing ', LOSS{l}]);
19 for r = 1:length(REG)
20 display(sprintf('\tREG=%d', REG(r)));
21 for b = 1:length(BATCH)
22 display(sprintf('\tB=%d', BATCH(b)));
23 for d = 1:length(DIAG)
24 display(sprintf('\tDiagonal=%d', DIAG(d)));
25 [W, Xi, D] = mlr_train(X, Yrel, 10e5, LOSS{l}, REG(r), DIAG(d), BATCH(b));
26 imagesc(W); drawnow;
27 % [W, Xi, D] = mlr_train(X, Yclass, 10e5, LOSS{l}, REG(r), DIAG(d), BATCH(b));
28 % imagesc(W); drawnow;
29 end
30 end
31 end
32 end
33
34 end