предсказать список значений признаков в кератах

Question

Я пытаюсь использовать predict_classes в модели keras, даже если форма ввода выглядит как требуется, функция выдает исключение

model = get_model()
flist = [10, 1.0, 0.0, 0.0, 1]
X = np.array(flist)
print(X.shape) # prints (5,)
model.predict_classes(X)

продолжает выдавать ошибку

ValueError: Error when checking input: expected dense_1_input to have shape (5,) but got array with shape (1,)

Answer 1

Форма Х должна быть, (Number_of_samples, input_dim).Используйте np.expand_dims.

X = np.expand_dims(X,axis=0)