Skip to content

Add 3D support to miss_forest_impute#1052

Open
sueoglu wants to merge 5 commits intomainfrom
longitudinal/issue-948
Open

Add 3D support to miss_forest_impute#1052
sueoglu wants to merge 5 commits intomainfrom
longitudinal/issue-948

Conversation

@sueoglu
Copy link
Copy Markdown
Collaborator

@sueoglu sueoglu commented Apr 16, 2026

fixes #948
Extends miss_forest_impute to handle 3D EHRData inputs

  • Removed @function_2D_only() decorator
  • If the input is 3D: flattens the layer to (n_obs * n_t, n_vars) (flattening along axis 0) before imputation and reshapes back to (n_obs, n_vars, n_t) afterwards
  • Added guard raising ValueError when input is 3D but no layer is specified
  • Added tests for 3D imputation

@sueoglu sueoglu marked this pull request as ready for review April 21, 2026 11:46
@sueoglu sueoglu requested a review from eroell April 21, 2026 11:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Longitudinal miss_forest_impute

1 participant