Skip to content
This repository was archived by the owner on Feb 24, 2025. It is now read-only.
This repository was archived by the owner on Feb 24, 2025. It is now read-only.

Replication issue: Train.py #2

Description

@joshuasteier

Hi there,
I'm having a replication issue after running train.py(after preprocessing):
Index(['admission_type', 'insurance', 'marital_status', 'race', 'LOS', 'blood',
'circulatory', 'congenital', 'digestive', 'endocrine', 'genitourinary',
'infectious', 'injury', 'mental', 'misc', 'muscular', 'neoplasms',
'nervous', 'pregnancy', 'prenatal', 'respiratory', 'skin', 'ICU',
'NICU'],
dtype='object')
Traceback (most recent call last):
File "train.py", line 37, in
model.fit(data, categorical_features, epochs = epoch)
File "/usr/local/lib/python3.8/dist-packages/ctgan/synthesizers/base.py", line 49, in wrapper
return function(self, *args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/ctgan/synthesizers/ctgan.py", line 294, in fit
self._validate_discrete_columns(train_data, discrete_columns)
File "/usr/local/lib/python3.8/dist-packages/ctgan/synthesizers/ctgan.py", line 279, in _validate_discrete_columns
raise ValueError(f'Invalid columns found: {invalid_columns}')
ValueError: Invalid columns found: {'hair_medium', 'anatom_site_general_challenge', 'hair_dense', 'hair_short', 'other', 'sex', 'black_frame', 'target', 'ruler_mark'}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions