IntClassNorm/plot_norm_fun.m at main · abhranildas/IntClassNorm · 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
function plot_norm_fun(mu,v,fun,prior,plot_color,varargin)

    if ~exist('prior','var')
        prior=1;
    end

    if ~exist('plot_color','var')
        colors=colororder;
        plot_color=colors(1,:);
    end

    dim=length(mu);

    if isstruct(fun)
        % plot distribution of q(x)
        if nnz(fun.q2) % if the quadratic term exists
            % q(x) ~ generalized chi-squared
            [w,k,lambda,s,m]=norm_quad_to_gx2_params(mu,v,fun);
            [f,~,x]=gx2pdf('full',w,k,lambda,s,m);
            area(x,prior*f,'facecolor',plot_color,'facealpha',0.4,'edgecolor',plot_color,'edgealpha',0.5,'linewidth',1)
        else % q(x) ~ normal
            mu_q=fun.q1'*mu+fun.q0;
            v_q=fun.q1'*v*fun.q1;
            plot_normal(mu_q,v_q,prior,plot_color);
        end
        if dim==1
            xlabel('$q(x)$','interpreter','latex');
        else
            xlabel('$q(\mbox{\boldmath $x$})$','interpreter','latex');
        end
    elseif isa(fun,'function_handle')
        xlims=norm_fun_inv([.01 .99],mu,v,fun,varargin{:});
        x=linspace(xlims(1),xlims(2),100);
        f=norm_fun_pdf(x,mu,v,fun,varargin{:});
        % p=norm_fun_cdf(x,mu,v,fun,varargin{:});
        % dx=x(2)-x(1);
        % f=diff(p)/dx;
        % x=x(1:end-1);
        area(x,prior*f,'facecolor',plot_color,'facealpha',0.4,'edgecolor',plot_color,'edgealpha',0.5,'linewidth',1)
        if dim==1
            xlabel('$f(x)$','interpreter','latex');
        else
            xlabel('$f(\mbox{\boldmath $x$})$','interpreter','latex');
        end
    end