Skip to content

[🐞] <title>PEMS系列数据集预处理dow与模型所需输入不符合 #219

Closed
@ZYFand-0517

Description

@ZYFand-0517

在staeformer_arch.py中,
self.dow_embedding = nn.Embedding(7, dow_embedding_dim)
dow_emb = self.dow_embedding(dow.long()),期望的输入应该类似这样
[3, 3, 6, 6, 6, 2, 4, 0, 4, 2, 2, 2, 6, 2, 0, 4]
与PEMS04数据预处理脚本产生的输入:
image
不符合,应该是0-6的离散值才符合吧

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't workingneeds-triagedfor issues raised to be triaged

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions