wolffd@0: function h=plotgauss2d(mu, Sigma) wolffd@0: % PLOTGAUSS2D Plot a 2D Gaussian as an ellipse with optional cross hairs wolffd@0: % h=plotgauss2(mu, Sigma) wolffd@0: % wolffd@0: wolffd@0: h = plotcov2(mu, Sigma); wolffd@0: return; wolffd@0: wolffd@0: %%%%%%%%%%%%%%%%%%%%%%%% wolffd@0: wolffd@0: % PLOTCOV2 - Plots a covariance ellipse with major and minor axes wolffd@0: % for a bivariate Gaussian distribution. wolffd@0: % wolffd@0: % Usage: wolffd@0: % h = plotcov2(mu, Sigma[, OPTIONS]); wolffd@0: % wolffd@0: % Inputs: wolffd@0: % mu - a 2 x 1 vector giving the mean of the distribution. wolffd@0: % Sigma - a 2 x 2 symmetric positive semi-definite matrix giving wolffd@0: % the covariance of the distribution (or the zero matrix). wolffd@0: % wolffd@0: % Options: wolffd@0: % 'conf' - a scalar between 0 and 1 giving the confidence wolffd@0: % interval (i.e., the fraction of probability mass to wolffd@0: % be enclosed by the ellipse); default is 0.9. wolffd@0: % 'num-pts' - the number of points to be used to plot the wolffd@0: % ellipse; default is 100. wolffd@0: % wolffd@0: % This function also accepts options for PLOT. wolffd@0: % wolffd@0: % Outputs: wolffd@0: % h - a vector of figure handles to the ellipse boundary and wolffd@0: % its major and minor axes wolffd@0: % wolffd@0: % See also: PLOTCOV3 wolffd@0: wolffd@0: % Copyright (C) 2002 Mark A. Paskin wolffd@0: wolffd@0: function h = plotcov2(mu, Sigma, varargin) wolffd@0: wolffd@0: if size(Sigma) ~= [2 2], error('Sigma must be a 2 by 2 matrix'); end wolffd@0: if length(mu) ~= 2, error('mu must be a 2 by 1 vector'); end wolffd@0: wolffd@0: [p, ... wolffd@0: n, ... wolffd@0: plot_opts] = process_options(varargin, 'conf', 0.9, ... wolffd@0: 'num-pts', 100); wolffd@0: h = []; wolffd@0: holding = ishold; wolffd@0: if (Sigma == zeros(2, 2)) wolffd@0: z = mu; wolffd@0: else wolffd@0: % Compute the Mahalanobis radius of the ellipsoid that encloses wolffd@0: % the desired probability mass. wolffd@0: k = conf2mahal(p, 2); wolffd@0: % The major and minor axes of the covariance ellipse are given by wolffd@0: % the eigenvectors of the covariance matrix. Their lengths (for wolffd@0: % the ellipse with unit Mahalanobis radius) are given by the wolffd@0: % square roots of the corresponding eigenvalues. wolffd@0: if (issparse(Sigma)) wolffd@0: [V, D] = eigs(Sigma); wolffd@0: else wolffd@0: [V, D] = eig(Sigma); wolffd@0: end wolffd@0: % Compute the points on the surface of the ellipse. wolffd@0: t = linspace(0, 2*pi, n); wolffd@0: u = [cos(t); sin(t)]; wolffd@0: w = (k * V * sqrt(D)) * u; wolffd@0: z = repmat(mu, [1 n]) + w; wolffd@0: % Plot the major and minor axes. wolffd@0: L = k * sqrt(diag(D)); wolffd@0: h = plot([mu(1); mu(1) + L(1) * V(1, 1)], ... wolffd@0: [mu(2); mu(2) + L(1) * V(2, 1)], plot_opts{:}); wolffd@0: hold on; wolffd@0: h = [h; plot([mu(1); mu(1) + L(2) * V(1, 2)], ... wolffd@0: [mu(2); mu(2) + L(2) * V(2, 2)], plot_opts{:})]; wolffd@0: end wolffd@0: wolffd@0: h = [h; plot(z(1, :), z(2, :), plot_opts{:})]; wolffd@0: if (~holding) hold off; end wolffd@0: wolffd@0: %%%%%%%%%%%% wolffd@0: wolffd@0: % CONF2MAHAL - Translates a confidence interval to a Mahalanobis wolffd@0: % distance. Consider a multivariate Gaussian wolffd@0: % distribution of the form wolffd@0: % wolffd@0: % p(x) = 1/sqrt((2 * pi)^d * det(C)) * exp((-1/2) * MD(x, m, inv(C))) wolffd@0: % wolffd@0: % where MD(x, m, P) is the Mahalanobis distance from x wolffd@0: % to m under P: wolffd@0: % wolffd@0: % MD(x, m, P) = (x - m) * P * (x - m)' wolffd@0: % wolffd@0: % A particular Mahalanobis distance k identifies an wolffd@0: % ellipsoid centered at the mean of the distribution. wolffd@0: % The confidence interval associated with this ellipsoid wolffd@0: % is the probability mass enclosed by it. Similarly, wolffd@0: % a particular confidence interval uniquely determines wolffd@0: % an ellipsoid with a fixed Mahalanobis distance. wolffd@0: % wolffd@0: % If X is an d dimensional Gaussian-distributed vector, wolffd@0: % then the Mahalanobis distance of X is distributed wolffd@0: % according to the Chi-squared distribution with d wolffd@0: % degrees of freedom. Thus, the Mahalanobis distance is wolffd@0: % determined by evaluating the inverse cumulative wolffd@0: % distribution function of the chi squared distribution wolffd@0: % up to the confidence value. wolffd@0: % wolffd@0: % Usage: wolffd@0: % wolffd@0: % m = conf2mahal(c, d); wolffd@0: % wolffd@0: % Inputs: wolffd@0: % wolffd@0: % c - the confidence interval wolffd@0: % d - the number of dimensions of the Gaussian distribution wolffd@0: % wolffd@0: % Outputs: wolffd@0: % wolffd@0: % m - the Mahalanobis radius of the ellipsoid enclosing the wolffd@0: % fraction c of the distribution's probability mass wolffd@0: % wolffd@0: % See also: MAHAL2CONF wolffd@0: wolffd@0: % Copyright (C) 2002 Mark A. Paskin wolffd@0: wolffd@0: function m = conf2mahal(c, d) wolffd@0: wolffd@0: m = chi2inv(c, d); % matlab stats toolbox