ciss_vae.data package

Module contents

Data loading utilities for CISS-VAE example dataset.

load_example_dataset()[source]

Load the complete CISS-VAE example dataset.

Returns:

A tuple containing (df_missing, df_complete, clusters) where: - df_missing : pd.DataFrame

DataFrame with missing values

  • df_completepd.DataFrame

    Complete DataFrame without missing values

  • clustersnp.ndarray

    Cluster labels as 1D array

Return type:

tuple

Examples

>>> from ciss_vae.data import load_example_dataset
>>> df_missing, df_complete, clusters = load_example_dataset()
load_missing_data()[source]

Load only the missing data DataFrame.

load_complete_data()[source]

Load only the complete data DataFrame.

load_clusters()[source]

Load only the cluster labels as 1D array.