def train_r(buffered_size=1024): def reader(): xs=[] ys=[] for i in range(1, 6): train_dict = unpickle("/home/aistudio/cifar-10-batches-py/data_batch_%d" % (i,)) xs.append(train_dict[b'data']) ys.append(train_dict[b'labels']) Xtr = np.concatenate(xs) Ytr = np.concatenate(ys) for (x, y) in zip(Xtr, Ytr): yield x, int(y) return paddle.reader.xmap_readers(train_mapper, reader, cpu_count(), buffered_size) # 对自定义数据集创建训练集test 的reader def test_r(buffered_size=1024): def reader(): test_dict = unpickle("/home/aistudio/cifar-10-batches-py/test_batch") X = test_dict[b'data'] Y = test_dict[b'labels'] for (x, y) in zip(X, Y): yield x, int(y) return paddle.reader.xmap_readers(test_mapper, reader, cpu_count(), buffered_size),这段程序如何改成我自己的数据集集啊,我数据集的路径在这儿D:\wjd\tpfl
时间: 2024-03-27 19:35:36 浏览: 50
INIT_I2C.rar_I2C dsp
你可以按照以下步骤修改代码中的数据集路径,以适应你自己的数据集:
1. 将数据集放置在你的电脑中,并记下数据集的路径。
2. 修改 train_r 函数中的数据集路径:
```python
train_dict = unpickle("数据集路径")
```
3. 修改 test_r 函数中的数据集路径:
```python
test_dict = unpickle("数据集路径")
```
4. 将修改后的代码保存并运行即可。
阅读全文