Ошибка при проверке ввода: ожидалось, что dens_54_input имеет shape (8,), но получил массив с shape (2,)

Question

Я получаю эту ошибку, Ошибка при проверке ввода: ожидалось, что dens_54_input будет иметь форму (8,), но получил массив с формой (2,), и я не знаю, почему ...

df = pd.read_csv('drain.csv')
values = df.values
X = df.iloc[:,:2]
Y = df.iloc[:,2:]
def create_model():
# create model
    model = Sequential()
    model.add(Dense(12, input_dim=2, activation='relu'))
    model.add(Dense(2, activation='relu'))
    model.add(Dense(1, activation='sigmoid'))
    # Compile model
    model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
    return model
seed = 7
numpy.random.seed(seed)
y_train = keras.utils.to_categorical(Y_train, 2)
y_test = keras.utils.to_categorical(Y_test, 2)

проблема возникает в этой строке:

history = model.fit(X_train, y_train,
                    batch_size=32,
                    epochs=15,
                    verbose=1,
                    validation_data=(X_test, y_test))