1import torch
2import torch.nn as nn
3
4
5class Model(nn.Module):
6    """
7    Custom model for short-term time series forecasting (M4 dataset).
8
9    Forward signature: forward(x_enc, x_mark_enc, x_dec, x_mark_dec, mask=None)
10    - x_enc: [batch, seq_len, enc_in] — input time series (enc_in=1 for M4)
11    - x_mark_enc: [batch, seq_len, time_features] — time feature encoding
12    - x_dec: [batch, label_len+pred_len, dec_in] — decoder input
13    - x_mark_dec: [batch, label_len+pred_len, time_features] — decoder time features
14
15    Must return: [batch, pred_len, c_out] for forecasting
... 27 more lines