unit8co/darts

Retrieve last hidden state for RNNModel and BlockRNNModel

Open

#1,649 创建于 2023年3月16日

在 GitHub 查看
 (2 评论) (1 反应) (0 负责人)Python (6,832 star) (762 fork)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
议题类型
feature
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
half day
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
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.
Retrieve last hidden state for RNNModel and BlockRNNModel · unit8co/darts#1649 | Good First Issue