(self, c_in, d_model, embed_type='fixed', freq='h', dropout=0.1)
| 137 | |
| 138 | class DataEmbedding_ms(nn.Module): |
| 139 | def __init__(self, c_in, d_model, embed_type='fixed', freq='h', dropout=0.1): |
| 140 | super(DataEmbedding_ms, self).__init__() |
| 141 | |
| 142 | self.value_embedding = TokenEmbedding(c_in=1, d_model=d_model) |
| 143 | self.position_embedding = PositionalEmbedding(d_model=d_model) |
| 144 | self.temporal_embedding = TemporalEmbedding(d_model=d_model, embed_type=embed_type, |
| 145 | freq=freq) if embed_type != 'timeF' else TimeFeatureEmbedding( |
| 146 | d_model=d_model, embed_type=embed_type, freq=freq) |
| 147 | self.dropout = nn.Dropout(p=dropout) |
| 148 | |
| 149 | def forward(self, x, x_mark): |
| 150 | B, T, N = x.shape |
nothing calls this directly
no test coverage detected