python - ValueError: 层dense_1的输入0与层不兼容
问题描述
我第一次使用 tensorflow,并将具有 18 个特征的数据分类为 4 个类。
X_train 的维度为:(14125,18)。
这是我的代码:
dataset = tf.data.Dataset.from_tensor_slices((np.array(X_train.values, dtype=float),
np.array(y_train.pet_category.values, dtype=float)))
train_data = dataset.shuffle(len(X_train)).batch(32)
vdataset = tf.data.Dataset.from_tensor_slices((np.array(X_val.values, dtype=float)))
val_data = vdataset.batch(32)
tfmodel = tf.keras.Sequential([
tf.keras.layers.Dense(15, activation=tf.nn.relu, input_shape=(18,1)),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(10, activation=tf.nn.relu),
tf.keras.layers.Dense(4, activation=tf.nn.softmax)
])
tfmodel.compile(optimizer='adam',
loss=tf.keras.losses.CategoricalCrossentropy(),
metrics=['accuracy'])
在调用tfmodel.fit(dataset, epochs=15, validation_data=val_data)
时,我收到以下错误:
ValueError: Input 0 of layer dense_1 is incompatible with the layer: expected axis -1 of input shape to have value 270 but received input with shape [18, 15]
我尝试寻找类似的问题,但找不到任何有用的东西。对解决这个问题真的很有帮助
编辑:问题出在版本上。当我使用较低版本的 TensorFlow(v 2.1.0)时,它就消失了。
解决方案
您正在使用dataset
intfit
而不是train_data
. X_train
我假设你正在使用一个名为and的 DataFrame y_train
,我用 numpy 模仿了它,它现在可以工作了。见下文。
import tensorflow as tf
import numpy as np
X_train = np.random.random((14125,18))
y_train = np.random.random((14125,1))
dataset = tf.data.Dataset.from_tensor_slices((X_train, y_train))
train_data = dataset.shuffle(len(X_train)).batch(32)
train_data = train_data.prefetch(
buffer_size=tf.data.experimental.AUTOTUNE)
tfmodel = tf.keras.Sequential([
tf.keras.layers.Dense(15, activation=tf.nn.relu, input_shape=(18,)),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(10, activation=tf.nn.relu),
tf.keras.layers.Dense(4, activation=tf.nn.softmax)
])
tfmodel.compile(optimizer='adam',
loss=tf.keras.losses.CategoricalCrossentropy(),
metrics=['accuracy'])
tfmodel.fit(train_data, epochs=5)
注意:我没有使用val_data
Train for 442 steps
Epoch 1/5
442/442 [==============================] - 1s 2ms/step - loss: 7.8375 - accuracy: 1.4159e-04
Epoch 2/5
442/442 [==============================] - 1s 2ms/step - loss: 28.5034 - accuracy: 0.0000e+00
Epoch 3/5
442/442 [==============================] - 1s 1ms/step - loss: 17.8604 - accuracy: 0.0000e+00
Epoch 4/5
442/442 [==============================] - 1s 1ms/step - loss: 3.4244 - accuracy: 2.1239e-04
Epoch 5/5
442/442 [==============================] - 1s 2ms/step - loss: 3.2791 - accuracy: 0.0160
<tensorflow.python.keras.callbacks.History at 0x7f0d8c72d630>
推荐阅读
- flutter - 按 fab 更改底部导航栏选定的项目
- hibernate - Bean 方法关闭会话导致没有会话或会话已关闭异常
- javascript - 获取 rowIndex 表格行中的按钮
- node.js - “由于平台响应无效,无法将 Dialogflow 响应解析为 AppResponse
- sql - Oracle 19c 在视图上触发 INSTEAD OF;CHAR 和 VARCHAR2 字段不会绑定 PLS-00049:错误的绑定变量 'NEW.component'
- wordpress - 在自定义感谢页面上显示 Woocommerce 订单详细信息
- javascript - 如何在 React 中遍历 JSON 对象
- powershell - 如何使用 DSC 配置 httpErrors defaultPath
- c# - C# ElasticSearch NEST,复合聚合脚本转换错误
- ios - 无法在 iOS 13 中设置标签栏阴影图像