unit8co/darts

Dirichlet Likelihood TFT model

Open

#1,813 建立於 2023年6月3日

在 GitHub 查看
 (6 留言) (0 反應) (0 負責人)Python (6,832 star) (762 fork)batch import
buggood first issue

描述

Hello, I'm tring to use DirichletLikelihood in TFT model, but I got this issue. How can I fix it?

`ValueError: Expected value argument (Tensor of shape (32, 12, 1)) to be within the support (Simplex()) of the distribution Dirichlet(concentration: torch.Size([12, 1])), but found invalid values:
tensor([[[0.5691],
         [0.8910],
         [0.6211],
         [0.0000],
         [0.3685],
         [0.3022],
         [0.5626],
         [0.3967],
         [0.6753],
         [0.6197],
         [0.4106],
         [0.6041]],
...
        [[0.0000],
         [0.0000],
         [0.0000],
         [0.0000],
         [0.0000],
         [0.0000],
         [0.4241],
         [0.3950],
         [0.3232],
         [0.2716],
         [0.3947],
         [0.2650]]])`

Thanks

貢獻者指南

技術棧
pythonpytorch
領域
datamachine learning
議題類型
bug
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 hours
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
needs investigation
前置要求
PyTorch basicsTemporal Fusion TransformerDirichlet distribution
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
40
研究方向
The error indicates that the TFT model output tensor of shape (32, 12, 1) is being passed as the value argument to Dirichlet, but Dirichlet expects a simplex (values summing to 1 along the last dimension). The issue likely stems from the TFT model's output not being normalized. Investigate the darts source code for the TFT model and its integration with DirichletLikelihood, specifically how the output is transformed before being passed to the distribution. Look at the comments in the issue for any additional context or attempted fixes. Check if there is a missing softmax or normalization step in the model head.