
收藏
#此函数只要稍作修改,就可以加载任意自己想要的数据格式。
def data_loader(data,seq_length,batch_size):
def reader():
data_list = []
label_list = []
for i in range(len(data)-seq_length):
data_list.append(data[i:i+seq_length,:-1])
label_list.append(data[i:i+seq_length,-1])
if len(data_list) == batch_size:
data_array = torch.tensor(np.array(data_list), dtype=torch.float)
label_array = torch.tensor(np.array(label_list), dtype=torch.float)
yield data_array, label_array
data_list = []
label_list = []
if len(data_list) > 0:
data_array = torch.tensor(np.array(data_list), dtype=torch.float)
label_array = torch.tensor(np.array(label_list), dtype=torch.float)
yield data_array, label_array
return reader
冷风
2021年9月10日 下午2:28
特别棒
小墨
2021年9月11日 上午9:15
谢谢
mobai
2021年9月10日 下午2:41
超级赞
不眠的夏天
2021年9月10日 下午2:43
代码很好用,已经珍藏
小墨
2021年9月11日 下午1:36
谢谢支持
uksully
2021年9月10日 下午2:48
简洁好用,支持作者
小墨
2021年9月11日 上午9:15
谢谢支持
墨白
2021年9月26日 下午9:28
已经收藏,确实不错,感谢作者的分享
小墨
2021年9月27日 上午11:04
谢谢支持