text_plain_output_1.png
train = pd.DataFrame(X, columns=[f'feat_{idx + 1}' for idx in range(X.shape[1])])
num_cols = train.columns.tolist()
categories = ['a'] * 50 + ['b'] * 25 + ['c'] * 25
random.shuffle(categories)
train['feat_5'] = categories
cat_cols = ['feat_5']

target = pd.DataFrame(y, columns=[f'target_{idx + 1}' for idx in range(y.shape[1])])
y_names = target.columns.tolist()
target.head(2)