出错代码行
model.fit(x_train, y_train, batch_size=32, epochs=50, validation_data=(x_test, y_test), validation_freq=1)
仔细研究,发现问题是在我数据集中。
我的数据集采用的都是int型表示灰度值,没有在数值后加.,而神经网络中需要输入的数据类型为float32,所以我们要用astype(np.float32)更改数据集数据类型。
出错代码:
all_inputs = iris_data[['1', '2', '3', '4', '5', '6', '7', '8', '9', '10']].values
all_classes = iris_data['Species'].values
修改为:
all_inputs = iris_data[['1', '2', '3', '4', '5', '6', '7', '8', '9', '10']].values.astype(np.float32)
all_classes = iris_data['Species'].values.astype(np.float32)