libcity.data.batch¶

class libcity.data.batch.Batch(feature_name)[source]¶

Bases: object

append(item)[source]¶

append a new item into the batch

Parameters: item (list) – 一组输入，跟feature_name的顺序一致，feature_name即是这一组输入的名字

to_ndarray()[source]¶

to_tensor(device)[source]¶

将数据self.data转移到device上

Parameters: device (torch.device) – GPU/CPU设备

class libcity.data.batch.BatchPAD(feature_name, pad_item=None, pad_max_len=None)[source]¶

Bases: libcity.data.batch.Batch

append(item)[source]¶

append a new item into the batch

Parameters: item (list) – 一组输入，跟feature_name的顺序一致，feature_name即是这一组输入的名字

get_origin_len(key)[source]¶

padding()[source]¶: 只提供对一维数组的特征进行补齐

to_tensor(device)[source]¶

将数据self.data转移到device上

Parameters: device (torch.device) – GPU/CPU设备