Mercurial > hg > camir-aes2014
diff reproduce_AES53rd/rerun_figure3.m @ 0:e9a9cd732c1e tip
first hg version after svn
author | wolffd |
---|---|
date | Tue, 10 Feb 2015 15:05:51 +0000 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/reproduce_AES53rd/rerun_figure3.m Tue Feb 10 15:05:51 2015 +0000 @@ -0,0 +1,39 @@ +% --- +% This script trains similarity measures and evaluates the +% impact of the number of hidden states as displayed in figure 3 of +% % Feature Preprocessing with RBMs for Music Similarity Learning +% Son N. Tran, Daniel Wolff, Tillman Weyde, Artur Garcez, AES53rd +% conference +% +% The output is printed in the console and plotted afterwards +% +% please note that the RBM training is a probabilistic process, and +% thus the papers' results can only be reproduced approximately with +% large numbers iterations of this script, and selection of RBMs according to +% their training set performance. +% --- + +% this version reproduces the figure approximately using precomputed RBM +[test(1), train(1)] = rbm_fig3('rbm_h30'); +[test(2), train(2)] = rbm_fig3('rbm_h50'); +[test(3), train(3)] = rbm_fig3('rbm_h100'); +[test(4), train(4)] = rbm_fig3('rbm_h500'); +[test(5), train(5)] = rbm_fig3('rbm_h1000'); + +% optionally, in order to test new RBMs, use the code below +% [test(1), train(1)] = rbm_fig3(30); +% [test(2), train(2)] = rbm_fig3(50); +% [test(3), train(3)] = rbm_fig3(100); +% [test(4), train(4)] = rbm_fig3(500); +% [test(5), train(5)] = rbm_fig3(1000); + +hidNum = [30 50 100 500 1000]; +hFig = figure; +set(hFig,'Units','centimeters'); +set(hFig, 'Position', [10 10 10 6]); +plot(hidNum,train*100,'--rx'); +hold on +plot(hidNum,test*100,'-bo'); +lg = legend('Training','Test'); +set(lg,'Location','SouthEast'); +title ('Figure 4: GRADIENT results for different hidNum');