Mercurial > hg > camir-aes2014
comparison reproduce_AES53rd/rerun_figure3/new_rbm.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 [mod] = new_rbm(hidNum,type) | |
2 % computes a new RBM that can be used instead of the saved ones | |
3 | |
4 conf.hidNum = hidNum; | |
5 conf.eNum = 150; | |
6 conf.bNum = 1; | |
7 conf.gNum = 1; | |
8 conf.N = 80; | |
9 conf.MAX_INC = 10; | |
10 | |
11 % use the specific parameters from the paper | |
12 switch type | |
13 | |
14 case 'svm' | |
15 conf.params = [0.7 0.7 0.1 2.0e-5]; | |
16 case 'grad' | |
17 conf.params = [0.05 0.1 0.1 2.0e-5]; | |
18 otherwise | |
19 end | |
20 | |
21 % load data | |
22 feature_file = 'rel_music_raw_features+simdata_ISMIR12'; | |
23 vars = whos('-file', feature_file); | |
24 A = load(feature_file,vars(1).name,vars(2).name,vars(3).name,vars(4).name); | |
25 data = A.(vars(1).name); | |
26 conf.sNum = size(data,1); | |
27 | |
28 mod = struct(); | |
29 [mod.W_max{1} mod.vB_max{1} mod.hB_max{1}] = training_rbm(conf,zeros(0,0),feature_file); | |
30 |