libcity.data.dataset.eta_dataset¶

class libcity.data.dataset.eta_dataset.ETADataset(config)[source]¶

_divide_data()[source]¶

_encode_traj(data)[source]¶

encode the trajectory

Parameters

data (dict) –

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

uid: [
trajectory1, trajectory2

]

} trajectory1 = [

checkin_record, checkin_record, …..

]

Returns

For example:

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

}

Return type

dict

_load_dyna()[source]¶

轨迹存储格式: (dict)

{

uid: [

], [

dyna_record, dyna_record, …

}

get_data()[source]¶

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

Returns

tuple contains:

train_dataloader: Dataloader composed of Batch (class)

eval_dataloader: Dataloader composed of Batch (class)

test_dataloader: Dataloader composed of Batch (class)

Return type

tuple

get_data_feature()[source]¶

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