Select the same number of samples from each class.

Question

DuckDuck 2018-7-6

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/409105-select-the-same-number-of-samples-from-each-class

评论： DuckDuck 2018-7-9

Say I have a table with 2 columns, one column stores the value the other the class of the value. The classes are imbalanced. I want to randomly select the same number of samples from each class. Is there any easy way to do it in Matlab.

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Jan 2018-7-6

Please post some Matlab code, which creates a small data set. I cannot guess, how you store "values" and "classes". What does "imbalanced" mean here?

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

KSSV 2018-7-6

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/409105-select-the-same-number-of-samples-from-each-class#answer_327769

在 MATLAB Online 中打开

 % create some random data for demo
  K = 3 ;  % number of groups 
  N = 5000 ;
  x = rand(N,1) ; 
  y = rand(N,1) ; 
  % apply kmeans to divide points into 3 groups 
  idx = kmeans([x,y],K) ;
A = [x y idx] ;  % the data with three classes 
% Get number of data in each class
ng1 = nnz(idx==1) ;   % number of points in group 1
ng2 = nnz(idx==2) ; % number of points in group 1
ng3 = nnz(idx==3) ; % number of points in group 1
% select data for each group 
N = min([ng1 ng2 ng3]) ;
iwant = cell(K,1) ;
for i = 1:K
    G=A(A(:,3)==i,:) ; % group data
    idx = randperm(size(G,1),N) ;
    iwant{i} = G(idx) ;
end