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设备