about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/KPMtools/partitionData.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/KPMtools/partitionData.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/KPMtools/partitionData.m')
-rw-r--r--sourcecodes/bnt-master/KPMtools/partitionData.m25
1 files changed, 25 insertions, 0 deletions
diff --git a/sourcecodes/bnt-master/KPMtools/partitionData.m b/sourcecodes/bnt-master/KPMtools/partitionData.m
new file mode 100644
index 00000000..3e5eb5fa
--- /dev/null
+++ b/sourcecodes/bnt-master/KPMtools/partitionData.m
@@ -0,0 +1,25 @@
+function  varargout = partitionData(Ndata, varargin)
+% PARTITIONDATA Partition a vector of indices into random sets
+% [a,b,c,...] = partitionData(N, 0.3, 0.2, 0.5, ...)
+%
+% Examples:
+% [a,b,c]=partitionData(105,0.3,0.2,0.5);
+% a= 1:30, b=32:52, c=52:105 (last bin gets all the left over)
+
+Npartitions = length(varargin);
+perm = randperm(Ndata);
+%perm = 1:Ndata;
+ndx = 1;
+for i=1:Npartitions
+  pc(i) = varargin{i};
+  Nbin(i) = fix(Ndata*pc(i));
+  low(i) = ndx;
+  if i==Npartitions
+    high(i) = Ndata;
+  else
+    high(i) = low(i)+Nbin(i)-1;
+  end
+  varargout{i} = perm(low(i):high(i));
+  ndx = ndx+Nbin(i);
+end
+