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

first hg version after svn

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/core/magnatagatune/mlr_unittest.m	Tue Feb 10 15:05:51 2015 +0000
@@ -0,0 +1,34 @@
+function mlr_unittest(X, Yrel)
+
+% Loss values to test
+    LOSS    = {'AUC', 'Prec@k', 'MAP', 'MRR', 'NDCG'};
+
+
+    % Regularization values to test
+    REG = [0,1,2,3];
+
+    % Batch sizes to test
+    BATCH = [0 1 5];
+
+    % Diagonal settings
+    DIAG = [0 1];
+
+    figure(1);
+    for l = 1:length(LOSS)
+        display(['Testing ', LOSS{l}]);
+        for r = 1:length(REG)
+            display(sprintf('\tREG=%d', REG(r)));
+            for b = 1:length(BATCH)
+                display(sprintf('\tB=%d', BATCH(b)));
+                for d = 1:length(DIAG)
+                    display(sprintf('\tDiagonal=%d', DIAG(d)));
+                    [W, Xi, D] = mlr_train(X, Yrel, 10e5, LOSS{l}, REG(r), DIAG(d), BATCH(b));
+                    imagesc(W); drawnow;
+%                     [W, Xi, D] = mlr_train(X, Yclass, 10e5, LOSS{l}, REG(r), DIAG(d), BATCH(b));
+%                     imagesc(W); drawnow;
+                end
+            end
+        end
+    end
+
+end