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 

点赞(0) 打赏

评论列表 共有 0 条评论

暂无评论

微信公众账号

微信扫一扫加关注

发表
评论
返回
顶部