unit8co/darts

Retrieve last hidden state for RNNModel and BlockRNNModel

Open

#1,649 opened on 2023年3月16日

GitHub で見る
 (2 comments) (1 reaction) (0 assignees)Python (6,832 stars) (762 forks)batch import
good first issueimprovement

説明

Is your feature request related to a current problem? Please describe. The last hidden state is an encoding of the information contained in the time series and is useful to retrieve in some projects.

Describe proposed solution Add a method that retrieve the last hidden state. Or maybe there is already an easy way to retrieve it? I couldn't find it.

コントリビューターガイド

技術スタック
pythonpytorch
領域
machine learning
Issue 種別
feature
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
3
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
half day
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
needs maintainer response
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
mostly clear
前提条件
PyTorch RNN basicsdarts model structure
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
40
調査方針
The issue asks to add a method to retrieve the last hidden state from RNNModel and BlockRNNModel. Start by examining the source code for these models in darts/models/forecasting/ directory, particularly the forward pass output. Check if there is already a way to access hidden states (e.g., via return state parameter). Look at similar implementations in PyTorch Lightning or other codebases. Since there are no comments from maintainers, propose a design in the issue before coding.