Mercurial > hg > camir-aes2014
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 |