ciss_vae.utils.helpers.get_imputed
- get_imputed(model, data_loader, device='cpu')[source]
Returns a ClusterDataset where originally missing values have been replaced with model reconstructions.
Processes the dataset through the trained VAE model to reconstruct missing values, including validation-masked entries. The returned dataset maintains the same structure as the original but with missing values filled in.
- Parameters:
model (nn.Module) – Trained VAE model
data_loader (torch.utils.data.DataLoader) – DataLoader for the original ClusterDataset
device (str, optional) – Torch device for computations, defaults to “cpu”
- Returns:
ClusterDataset with reconstructed values filled in at originally missing positions
- Return type: