import mathimport randomclasses=750samples = datashape0perClass = intsamplesclassesselected = intperClass07trainDatatestData = trainLabeltestLabel = count1 = for i in rangeclasses count2k = mathinf
这段代码通过随机选择样本来划分训练集和测试集。首先,它将每个类别中的样本数量除以总类别数,得到每个类别应该有的样本数量。然后,它选择70%的样本作为训练集,30%的样本作为测试集。在选择训练集样本时,它使用一个计数器count2来记录已选择的样本的索引,然后随机生成一个索引k,确保k不在count2中,将该样本添加到训练集中,并将其索引添加到count1中。最后,通过将数据和标签转换为矩阵的形式,返回划分好的训练集数据、训练集标签、测试集数据和测试集标签。
原文地址: https://www.cveoy.top/t/topic/h5hm 著作权归作者所有。请勿转载和采集!