BNP-Track/sampler_update.m at main · LabPresse/BNP-Track · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
function sample = sampler_update( sample, params )

%% update counter
sample.i = sample.i + 1;
sample.F = get_F(sample.i,params);

%% random sweep
for tag = randperm(9)
    switch tag
        case 1
            [sample.C,sample.rec(:,tag)] = sampler_update_BCK( ...
             sample.C,sample.h,sample.b,sample.X,sample.Y,sample.Z,sample.F,sample.rec(:,tag),params);
        case 2
            [sample.C,sample.h,sample.rec(:,tag)] = sampler_update_HHH( ...
             sample.C,sample.h,sample.rec(:,tag), ...
             sample.b,sample.X,sample.Y,sample.Z,sample.F,params);
        case 3
            [sample.b,sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_AAA( ...
             sample.b,sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.C,sample.h,sample.F,params);
        case 4
            [sample.b,sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_BBB( ...
             sample.b,sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.C,sample.h,sample.F,params);
        case 5
            [sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_PP0( ...
             sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.b,sample.C,sample.h,sample.F,params);
        case 6
            [sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_PPF( ...
             sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.C,sample.h,sample.b,sample.F,params);
        case 7
            [sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_ZZZ( ...
             sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.b,params);
        case 8
            [sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_ZZT( ...
             sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.b,params);
        case 9
            [sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag)] = sampler_update_LLL( ...
             sample.K,sample.X,sample.Y,sample.Z,sample.rec(:,tag), ...
             sample.b,params);
        otherwise
            error('Unknown sampler requested!')
    end
end

%% book-keeping
sample.L = get_log_probs(sample.F,sample.C,sample.b,sample.h,sample.K,sample.X,sample.Y,sample.Z,params);
sample.D = get_D(sample.X,sample.Y,sample.Z,sample.b,params);