我无法让 Keras 预测任何事情。即使在这个简约的模型中也不行:
from keras.models import Sequential
from keras.layers import Dense
import numpy as np
inDim = 3
outDim = 1
model = Sequential()
model.add(Dense(5, input_dim=inDim, activation='relu'))
model.add(Dense(outDim, activation='sigmoid'))
model.compile(loss='mse', optimizer='adam', metrics=['accuracy'])
test_input = np.zeros((1,inDim))
test_output = np.zeros((1,outDim))
model.fit(test_input, test_output)
prediction = model.predict(test_input)
一切都按预期进行,直到最后一行:
Epoch 1/1
1/1 [==============================] - 0s 448ms/step - loss: 0.2500 - acc: 1.0000
Traceback (most recent call last):
File "<ipython-input-24-ee244a6c7287>", line 16, in <module>
prediction = model.predict(test_input)
File "E:\Programme\Anaconda3\lib\site-packages\keras\engine\training.py", line 1172, in predict
steps=steps)
File "E:\Programme\Anaconda3\lib\site-packages\keras\engine\training_arrays.py", line 304, in predict_loop
outs.append(np.zeros(shape, dtype=batch_out.dtype))
TypeError: data type not understood
我一遍又一遍地尝试不同的数组和列表组合,但要么出现 TypeError,要么出现 ValueError,因为形状错误。
几个答案(例如here https://stackoverflow.com/questions/39950311/keras-error-on-predict)建议使用类似的东西
model.predict(np.array([[0,0,0]]))
但这对我也不起作用。
谁能告诉我如何正确地做到这一点?
编辑:显然,代码不是问题,见下文。