Open Junyanzhang opened 1 year ago
q_num = np.load(path + 'test_num.npy') v_num = np.load(path + 'val_num.npy') t_num = np.load(path + 'train_num.npy') d_num = np.load(path + 'database_num.npy')
抱歉,我并没有使用原文的数据读取方式。我在原文的代码基础上,加了一点数据划分的代码。具体实现方式是,每个类别各抽相同数量的数据,构建一个分布均衡的数据集。
q_num = np.load(path + 'test_num.npy') v_num = np.load(path + 'val_num.npy') t_num = np.load(path + 'train_num.npy') d_num = np.load(path + 'database_num.npy')