Mercurial > hg > camir-aes2014
annotate 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 |
rev | line source |
---|---|
wolffd@0 | 1 function mlr_unittest(X, Yrel) |
wolffd@0 | 2 |
wolffd@0 | 3 % Loss values to test |
wolffd@0 | 4 LOSS = {'AUC', 'Prec@k', 'MAP', 'MRR', 'NDCG'}; |
wolffd@0 | 5 |
wolffd@0 | 6 |
wolffd@0 | 7 % Regularization values to test |
wolffd@0 | 8 REG = [0,1,2,3]; |
wolffd@0 | 9 |
wolffd@0 | 10 % Batch sizes to test |
wolffd@0 | 11 BATCH = [0 1 5]; |
wolffd@0 | 12 |
wolffd@0 | 13 % Diagonal settings |
wolffd@0 | 14 DIAG = [0 1]; |
wolffd@0 | 15 |
wolffd@0 | 16 figure(1); |
wolffd@0 | 17 for l = 1:length(LOSS) |
wolffd@0 | 18 display(['Testing ', LOSS{l}]); |
wolffd@0 | 19 for r = 1:length(REG) |
wolffd@0 | 20 display(sprintf('\tREG=%d', REG(r))); |
wolffd@0 | 21 for b = 1:length(BATCH) |
wolffd@0 | 22 display(sprintf('\tB=%d', BATCH(b))); |
wolffd@0 | 23 for d = 1:length(DIAG) |
wolffd@0 | 24 display(sprintf('\tDiagonal=%d', DIAG(d))); |
wolffd@0 | 25 [W, Xi, D] = mlr_train(X, Yrel, 10e5, LOSS{l}, REG(r), DIAG(d), BATCH(b)); |
wolffd@0 | 26 imagesc(W); drawnow; |
wolffd@0 | 27 % [W, Xi, D] = mlr_train(X, Yclass, 10e5, LOSS{l}, REG(r), DIAG(d), BATCH(b)); |
wolffd@0 | 28 % imagesc(W); drawnow; |
wolffd@0 | 29 end |
wolffd@0 | 30 end |
wolffd@0 | 31 end |
wolffd@0 | 32 end |
wolffd@0 | 33 |
wolffd@0 | 34 end |