The RANDDIRICHLET function is part of the IMLMLIB library. The RANDDIRICHLET function generates a random sample from a Dirichlet distribution, which is a multivariate generalization of the beta distribution.
The input parameters are as follows:
is the number of observations to sample.
is a vector of shape parameters for the distribution, .
The RANDDIRICHLET function returns an matrix that contains random draws from the Dirichlet distribution.
If with and follows a Dirichlet distribution with shape parameter , then
the probability density function for is
if , the probability distribution is a beta distribution.
if , then
the expected value of is .
the variance of is .
the covariance of and is .
The following example generates 1,000 samples from a two-dimensional Dirichlet distribution. Each row of the returned matrix
x
is a row vector sampled from the Dirichlet distribution. The following example computes the sample mean and covariance and
compares them with the expected values:
call randseed(1); n = 1000; Shape = {2, 1, 1}; x = RandDirichlet(n,Shape); d = nrow(Shape)-1; s = Shape[1:d]; Shape0 = sum(Shape); Mean = s`/Shape0; Cov = -s*s` / (Shape0##2*(Shape0+1)); /* replace diagonal elements with variance */ Variance = s#(Shape0-s) / (Shape0##2*(Shape0+1)); do i = 1 to d; Cov[i,i] = Variance[i]; end; SampleMean = mean(x); SampleCov = cov(x); print SampleMean Mean, SampleCov Cov;
Figure 24.296: Estimated Mean and Covariance Matrix
SampleMean | Mean | ||
---|---|---|---|
0.4992449 | 0.2485677 | 0.5 | 0.25 |
SampleCov | Cov | ||
---|---|---|---|
0.0502652 | -0.026085 | 0.05 | -0.025 |
-0.026085 | 0.0393922 | -0.025 | 0.0375 |
For further details about sampling from the Dirichlet distribution, see Kotz, Balakrishnan, and Johnson (2000); Gentle (2003); or Devroye (1986).