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'}