简单的神经网络

Question

我有输入和输出（ XNOR 门），当我想训练它们时出现错误。这是代码：

import tensorflow as tf
import numpy as np

training_inputs = np.array([[0,0,0],[0,0,1],[0,1,0],[0,1,1],[1,0,0],[1,0,1],[1,1,0],[1,1,1]],dtype=float)
training_outputs =np.array([1,0,0,1,0,1,1,0],dtype=float)

model = tf.keras.Sequential([
  tf.keras.layers.Dense(units=1, input_shape=[1])
])


model.compile(loss='mean_squared_error',
              optimizer=tf.keras.optimizers.Adam(0.1))

history = model.fit(training_inputs, training_outputs , epochs=500, verbose=False)

错误：

ValueError: Exception encountered when calling layer "sequential_14" (type Sequential).
    
    Input 0 of layer "dense_14" is incompatible with the layer: expected axis -1of input shape to have value 1, but received input with shape (None, 2)

Answer 1

您的input_shape不正确。由于 training_inputs 具有 (8, 3) 的形状，这意味着 8 个样本，每个样本具有 3 个特征，因此您的模型应如下所示：

import tensorflow as tf
import numpy as np

training_inputs = np.array([[0,0,0],[0,0,1],[0,1,0],[0,1,1],[1,0,0],[1,0,1],[1,1,0],[1,1,1]],dtype=float)
training_outputs =np.array([1,0,0,1,0,1,1,0],dtype=float)

model = tf.keras.Sequential([
  tf.keras.layers.Dense(units=1, input_shape=(3,))
])


model.compile(loss='mean_squared_error',
              optimizer=tf.keras.optimizers.Adam(0.1))

history = model.fit(training_inputs, training_outputs , epochs=500, verbose=False, batch_size=2)

简单的神经网络

Simple Neural Network

python

machine-learning

neural-network

deep-learning

tensorflow