Error: Index 3 is out of bounds for axis 1 with size 2

Question

I am performing a classification task using CategoricalNB from sklearn. The dataframe is composed of categorical data and I split it such that:

X_train.shape