我有一个数据矩阵X(60×208)和一个标签矩阵Y(1×208)。我想将我的数据矩阵X随机分成两组列向量子集:训练集(占数据的70%)和测试集(占数据的30%),但我需要能够识别Y中的哪个标签对应于每个列向量。我找不到任何函数来完成这个任务,有什么建议吗?
编辑:我想补充一点,Y中只有两个标签:1和2(不知道这是否有影响)
回答:
这做起来相当简单。使用randperm
来生成从1
到你拥有的点数(在你的例子中是208)的随机排列索引。
一旦你生成了这个序列,只需使用它来从你的X
和Y
中提取训练和测试数据及标签。具体操作如下:
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));
split_point
决定了我们需要将多少点放入训练集中,我们需要对其进行四舍五入,以防计算结果有小数点。我也没有在代码中硬编码208,因为你的数据集可能会增长,所以这将适用于你选择的任何大小数据集。X_train
和Y_train
将包含你的训练集数据和标签,而X_test
和Y_test
将包含你的测试集数据和标签。
因此,X_train
的第一列是你训练集的第一个数据点,Y_train
的第一个元素是该点的标签…依此类推!