Keras model.predict不接受大小为1的输入(标量数)

时间:2016-09-13 04:15:38

标签: python keras

我是Keras和python的新手,现在我正在使用Keras找到数据模型并使用该model.predict进行优化,但是model.predict只能将输入视为numpy数组至少2个元素。

我的代码是

import keras
from keras.models import Sequential
from keras.layers import Dense
from keras.optimizers import SGD
import numpy as np

x = np.arange(-2,3.0,0.01)
y = x**2 - 2*x + 1

model = Sequential()
model.add(Dense(50, activation='sigmoid', 
                input_dim=1, init='uniform'))
model.add(Dense(1, activation='linear'))
sgd = SGD(lr=0.05, decay=1e-6, momentum=0.9, nesterov=False)
model.compile(loss='mean_squared_error', 
              optimizer='sgd',
              metrics=['accuracy'])
model.fit(x,y,nb_epoch=300, batch_size = 5,verbose = 0)

代码可以很好地适应,但是如果我尝试使用model.predict作为标量数字它会给我错误

(Pdb) model.predict(0.0)
*** Exception: Error when checking : data should be a Numpy array, or list/dict of Numpy arrays. Found: 0.0...

我强迫它成为numpy数组但仍然失败,它说输入需要是2维!

(Pdb) model.predict(np.asarray(0.0))
*** Exception: Error when checking : expected dense_input_1 to have 2 dimensions, but got array with shape ()

但如果我输入两个数字,那么它会给我答案

(Pdb) model.predict([0.0,0.0])
array([[ 1.07415712],
       [ 1.07415712]], dtype=float32)

我需要model.predict将单个数字作为输入用于优化。我不确定我使用的任何设置是错误的。请帮助,谢谢。

1 个答案:

答案 0 :(得分:1)

尝试:

model.predict(np.asarray(0.0).reshape((1,1)))

在Keras中,第一个维度始终与示例编号相关联,因此必须提供。