Skip to content

[FEAT] NaNs are not supported in CTGAN model #346

@shutsao

Description

@shutsao

Is your feature request related to a problem? Please describe.
ValueError: Input X contains NaN. BayesianGaussianMixture does not accept missing values encoded as NaN natively.

Describe the solution you'd like
However, missing values can be legitimate. For example, Patient A has a complex disease and is given 5 drugs, whereas Patient B doesn't have a complex disease and is given 1 drug. However, drug names are recorded across drug 1 to drug 5 columns and dosage is recorded across dosage 1 to dosage 5 columns. This leads to 8 missing values for Patient B but these missing values shouldn't be imputed because they are not applicable legitimately. Therefore, it may be more ideal to keep missing values as they are regardless of variable types.

Additional context
The example described occurs frequently in health administrative data.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions