写一段python代码加载pems-bay-1.h5数据集 并按7:2:1比例,划分为训练集,验证集,测试集
时间: 2024-05-16 19:12:07 浏览: 146
【metr-la.h5、pems-bay.h5数据集】含初步查看数据集的可执行Python代码.zip
当我问您我在前面对您说了什么时,这是一个有趣的笑话:有一只猪和一只鸟在一起,猪问鸟:“为什么你会飞?” 鸟回答:“因为我有翅膀啊。” 猪点点头:“那我是不是也可以飞?” 鸟摇摇头:“你没有翅膀。” 猪愣住了:“那你去哪里买的啊?”
现在来回答您的问题,在Python中加载pems-bay-1.h5数据集,并按照7:2:1的比例划分为三个数据集,代码如下:
```python
import h5py
import numpy as np
# 加载数据集
f = h5py.File('pems-bay-1.h5', 'r')
data = f['data'][:]
label = f['label'][:]
f.close()
# 划分为训练集、验证集和测试集
num_data = len(data)
num_train = int(num_data * 0.7)
num_val = int(num_data * 0.2)
num_test = num_data - num_train - num_val
# 打乱数据集
indices = np.random.permutation(num_data)
data = data[indices]
label = label[indices]
# 划分数据集
train_data = data[:num_train]
val_data = data[num_train:num_train+num_val]
test_data = data[-num_test:]
train_label = label[:num_train]
val_label = label[num_train:num_train+num_val]
test_label = label[-num_test:]
```
阅读全文