Matlab: How can I split my data matrix into two random subsets of column vectors while keeping the label information?

与世无争的帅哥 提交于 2020-01-19 05:37:30

问题


I have a data matrix X (60x208) and a matrix of labels Y (1x208). I want to split my data matrix X into two random subsets of column vectors: training (which will be 70% of the data) and testing (which will be 30% of the data), but I need to still be able to identify which label from Y corresponds to each column vector. I couldn't find any function to do this, any ideas?

EDIT: Thought I should add, there are only two labels in Y: 1 and 2 (not sure if this makes a difference)


回答1:


That's pretty easy to do. Use randperm to generate a random permutation of indices from 1 up to as many points as you have... which is 208 in your case.

Once you generate this sequence, simply use this and subset into your X and Y to extract the training and test data and labels. As such, do something like this:

num_points = size(X,2);
split_point = round(num_points*0.7);
seq = randperm(num_points);
X_train = X(:,seq(1:split_point));
Y_train = Y(seq(1:split_point));
X_test = X(:,seq(split_point+1:end));
Y_test = Y(seq(split_point+1:end));

The split_point determines how many points we need to place into our training set, and we will need to round it in case this calculation yields any decimal points. I also didn't hard code 208 in there because your data set might grow and so this will work with any size data set you choose. X_train and Y_train will contain your data and labels for your training set while X_test and Y_test will contain your data and labels for your test set.

As such, the first column of X_train is your data point for the first element of your training set, with the first element of Y_train serving as the label for that particular point... and so on and so forth!



来源:https://stackoverflow.com/questions/26662123/matlab-how-can-i-split-my-data-matrix-into-two-random-subsets-of-column-vectors

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!