ciss_vae.utils.helpers.get_imputed_df
- get_imputed_df(model, data_loader, device='cpu')[source]
Given trained model and cluster dataset object, get imputed dataset as pandas DataFrame.
Reconstructs missing values using the trained VAE model and returns the complete dataset with original scaling restored and validation entries replaced with true values.
- Parameters:
model (CISSVAE) – Trained CISSVAE model (should be in eval() mode)
data_loader (torch.utils.data.DataLoader) – DataLoader for the original ClusterDataset
device (str, optional) – Device to run computations on, defaults to “cpu”
- Returns:
DataFrame containing imputed (unscaled) data with original row ordering
- Return type: