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