python - 多变量/特征 Tensorflow

Question

我想使用多层keras对人的高度进行分类，如果高度大于170应返回1，而小于170应返回0

data_input = [[0,165], [0,166], [0,167], [0,172], [0,173]]
data_output = [0,0,0,1,1]
model = keras.Sequential([
  keras.layers.Flatten(input_shape=(1,)),
  keras.layers.Dense(2, activation=tf.nn.relu),
  keras.layers.Dense(2, activation=tf.nn.softmax)
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(data_input, data_output, epochs=5)
predictions = model.predict([0,184])
print(predictions)

但它给了我错误“ValueError:输入数组应具有与目标数组相同数量的样本。找到 2 个输入样本和 5 个目标样本。”

Answer 1

你应该使用 Numpy 来包装你的输入

data_input = np.array(data_input).reshape(5 ,2)
data_output = np.array(data_output).reshape(5)

顺便问一下，为什么使用 ML 来计算简单的条件分类问题