libcity.data.dataset.eta_dataset¶

class libcity.data.dataset.eta_dataset.ETADataset(config)[源代码]¶

_divide_data()[源代码]¶

_encode_traj(data)[源代码]¶

encode the trajectory

参数

data (dict) –

the key is uid, the value is the uid’s trajectories. For example: {

uid: [
trajectory1, trajectory2

]

} trajectory1 = [

checkin_record, checkin_record, …..

]

返回

For example:

{: data_feature: {…}, pad_item: {…}, encoded_data: {uid: encoded_trajectories}

}

返回类型

dict

_load_dyna()[源代码]¶

轨迹存储格式: (dict)

{

uid: [

], [

dyna_record, dyna_record, …

}

返回数据的DataLoader，包括训练数据、测试数据、验证数据

返回

tuple contains:

train_dataloader: Dataloader composed of Batch (class)

eval_dataloader: Dataloader composed of Batch (class)

test_dataloader: Dataloader composed of Batch (class)

返回类型

tuple

get_data_feature()[源代码]¶

返回一个 dict，包含数据集的相关特征