1 train_task1.py

1.1 main部分

读取命令行参数，调用task1函数

1.2 task1 train

1.3 task1 valid

1.3 collate_fn

2 Dataset

2.1 train dataset

 

2.2 valid dataset

3 LPBERT

3.1 不同的embedding

day-of-week embedding和time-of-day embedding

X位置和Y位置的embedding

时间间隔的embedding

3.2  Embedding Layer

五种embedding聚合

3.3   TransformerEncoderModel

3.4 Output layer

3.5 LPBERT

3.5