Dirichlet分布から疑似乱数ベクトルを生成

C言語によるサンプルソースコード : 使用関数名:nag_rand_dirichlet (g05sec)

ホーム > 製品 > nAG数値計算ライブラリ > サンプルソースコード集 > Dirichlet分布から疑似乱数ベクトルを生成 (C言語/C++)

Keyword: Dirichlet分布, 疑似乱数ベクトル

概要

本サンプルはDirichlet分布から疑似乱数ベクトルの生成を行うC言語によるサンプルプログラムです。 本サンプルは以下に示される確率密度関数で m(次元)=4 かつ α={2.0,2.0,2.0,2.0} の場合のDirichlet分布から5個の疑似乱数を生成し出力します。

Dirichlet分布のデータ 

※本サンプルはnAG Cライブラリに含まれる関数 nag_rand_dirichlet() のExampleコードです。本サンプル及び関数の詳細情報は nag_rand_dirichlet のマニュアルページをご参照ください。
ご相談やお問い合わせはこちらまで

出力結果

(本関数の詳細はnag_rand_dirichlet のマニュアルページを参照)
1
2
3
4
5
6
7

この出力例をダウンロード
nag_rand_dirichlet (g05sec) Example Program Results

    0.3600    0.3138    0.0837    0.2426
    0.2874    0.5121    0.1497    0.0509
    0.2286    0.2190    0.3959    0.1566
    0.1744    0.3961    0.2764    0.1530
    0.1522    0.2845    0.2074    0.3559

  • 3~7行目に生成された疑似乱数が出力されています。

ソースコード

(本関数の詳細はnag_rand_dirichlet のマニュアルページを参照)

※本サンプルソースコードはnAG数値計算ライブラリ(Windows, Linux, MAC等に対応)の関数を呼び出します。
サンプルのコンパイル及び実行方法

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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

このソースコードをダウンロード
/* nag_rand_dirichlet (g05sec) Example Program.
 *
 * CLL6I261D/CLL6I261DL Version.
 *
 * Copyright 2017 Numerical Algorithms Group.
 *
 * Mark 26.1, 2017.
 */
/* Pre-processor includes */
#include <stdio.h>
#include <math.h>
#include <nag.h>
#include <nag_stdlib.h>
#include <nagg05.h>

#define X(I, J) x[(order == Nag_ColMajor)?(J*pdx + I):(I*pdx + J)]

int main(void)
{
  /* Integer scalar and array declarations */
  Integer exit_status = 0;
  Integer pdx, x_size, i, j, lstate;
  Integer *state = 0;

  /* nAG structures */
  NagError fail;

  /* Double scalar and array declarations */
  double *x = 0;

  /* Set the distribution parameters */
  double a[] = { 2.0e0, 2.0e0, 2.0e0, 2.0e0 };
  Integer m = 4;

  /* Set the sample size */
  Integer n = 5;

  /* Return the results in column major order */
  Nag_OrderType order = Nag_ColMajor;

  /* Choose the base generator */
  Nag_BaseRNG genid = Nag_Basic;
  Integer subid = 0;

  /* Set the seed */
  Integer seed[] = { 1762543 };
  Integer lseed = 1;

  /* Initialize the error structure */
  INIT_FAIL(fail);

  printf("nag_rand_dirichlet (g05sec) Example Program Results\n\n");

  /* Get the length of the state array */
  lstate = -1;
  nag_rand_init_repeatable(genid, subid, seed, lseed, state, &lstate, &fail);
  if (fail.code != NE_NOERROR) {
    printf("Error from nag_rand_init_repeatable (g05kfc).\n%s\n",
           fail.message);
    exit_status = 1;
    goto END;
  }

  pdx = (order == Nag_ColMajor) ? n : m;
  x_size = (order == Nag_ColMajor) ? pdx * m : pdx * n;

  /* Allocate arrays */
  if (!(x = nAG_ALLOC(x_size, double)) ||
      !(state = nAG_ALLOC(lstate, Integer)))
  {
    printf("Allocation failure\n");
    exit_status = -1;
    goto END;
  }

  /* Initialize the generator to a repeatable sequence */
  nag_rand_init_repeatable(genid, subid, seed, lseed, state, &lstate, &fail);
  if (fail.code != NE_NOERROR) {
    printf("Error from nag_rand_init_repeatable (g05kfc).\n%s\n",
           fail.message);
    exit_status = 1;
    goto END;
  }

  /* Generate the variates */
  nag_rand_dirichlet(order, n, m, a, state, x, pdx, &fail);
  if (fail.code != NE_NOERROR) {
    printf("Error from nag_rand_dirichlet (g05sec).\n%s\n", fail.message);
    exit_status = 1;
    goto END;
  }

  /* Display the variates */
  for (i = 0; i < n; i++) {
    for (j = 0; j < m; j++)
      printf("%10.4f", X(i, j));
    printf("\n");
  }

END:
  nAG_FREE(x);
  nAG_FREE(state);

  return exit_status;
}


関連情報
© 日本ニューメリカルアルゴリズムズグループ株式会社 2024
Privacy Policy  /  Trademarks