Jane Street 실시간 시장 데이터 예측 대회 우승 솔루션 분석

class ParamModel(nn.Module):
    def __init__(self, config):
        super().__init__()
        self.config = config
        self.rnn_multiplier = config.get('rnn_multiplier', 1)
        # ... (중략)
        self.asset_mixer = AxialAttention(dim, axis=1)
        self.time_mixer = nn.GRU(dim * self.rnn_multiplier, dim * self.rnn_multiplier, batch_first=True)

    def forward(self, x, h=None):
        # x shape: (batch, time, asset, feature)
        x = self.asset_mixer(x)
        x = x.view(batch * asset, time, -1)
        x, h = self.time_mixer(x, h)
        return x, h