Hiperparâmetros
Max prediction length: 54 meses
Max encoder length: 542
Min encoder length: 271
Batch size: 128
Patience: 10 épocas
Hidden size: 16
Attention head size: 1
Drop out: 0.1
Hidden continuous size: 8
Output size: 7 quantis
Log interval: 10 batches
Função de perda: perda por quantil
Épocas: 500
Detalhes
Prevenção de overfitting: EarlyStopping do PyTorch Lightning
Taxa de aprendizagem: Tuner do PyTorch Lightning
Log de informações: TensorBoardLogger do TensorFlow
Variáveis de entrada categóricas conhecidas no futuro: mês