text_plain_output_1.png
import pandas as pd

train = pd.read_csv('../input/30-days-of-ml/train.csv')
test = pd.read_csv('../input/30-days-of-ml/test.csv')

X_train = train.drop(['id', 'target'], axis=1)
y_train = train['target']
X_test = test.drop(['id'], axis=1)

cat_features = [i for i, col in enumerate(X_train.columns) if 'cat' in col]
cat_features