- train_x = []
- val_x = []
- test_x = []
- train_y = []
- val_y = []
- test_y = []
- samples = np.split(data, 100)
- targets = np.split(labels, 100)
- for sample in samples:
- train_x.extend(sample[0:12])
- val_x.extend(sample[12:17])
- test_x.extend(sample[17:24])
- for target in targets:
- train_y.extend(target[0:12])
- val_y.extend(target[12:17])
- test_y.extend(target[17:24])
- print(len(train_x), len(val_y), len(test_x))
- print(len(train_y), len(val_y), len(test_y))
- print(train_x)
- View Code
来源: http://www.bubuko.com/infodetail-3158831.html