about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/KPMstats/sample.m
diff options
context:
space:
mode:
authorziejd22018-03-14 23:23:33 -0500
committerGitHub2018-03-14 23:23:33 -0500
commit1ff6baa44e22b91eefb48aea6f3befa078c0489b (patch)
treee0fd79d2e32fd2aedda2eadaed0f19af3514c520 /sourcecodes/bnt-master/KPMstats/sample.m
parent6882395afdadf4e982b25b5215071a0932730950 (diff)
parentc80226899f5cdd9f11c163817d59445213f5bef0 (diff)
downloadBNW-1ff6baa44e22b91eefb48aea6f3befa078c0489b.tar.gz
Merge pull request #1 from ziejd2/octave_php_separate
Octave php separate
Diffstat (limited to 'sourcecodes/bnt-master/KPMstats/sample.m')
-rw-r--r--sourcecodes/bnt-master/KPMstats/sample.m15
1 files changed, 15 insertions, 0 deletions
diff --git a/sourcecodes/bnt-master/KPMstats/sample.m b/sourcecodes/bnt-master/KPMstats/sample.m
new file mode 100644
index 00000000..fc252da2
--- /dev/null
+++ b/sourcecodes/bnt-master/KPMstats/sample.m
@@ -0,0 +1,15 @@
+function x = sample(p, n)
+% SAMPLE    Sample from categorical distribution.
+% Returns a row vector of integers, sampled according to the probability
+% distribution p.
+% Uses the stick-breaking algorithm.
+% Much faster algorithms are also possible.
+
+if nargin < 2
+  n = 1;
+end
+
+cdf = cumsum(p(:));
+for i = 1:n
+  x(i) = sum(cdf < rand) + 1;
+end