Error: Error al verificar la entrada: se esperaba que dense_Dense1_input tuviera 3 dimensiones. pero tengo una matriz con forma 1,9
Soy realmente nuevo en tensorflow.js y estoy tratando de hacer un modelo simple que te diga en qué lado del lienzo hiciste clic
const model = tf.sequential();
model.add(
tf.layers.dense({
units: 200,
activation: "sigmoid",
inputShape: [0, 1],
})
);
model.add(
tf.layers.dense({
units: 2,
activation: "softmax",
})
);
model.compile({
optimizer: optimizer,
loss: "categoricalCrossentropy",
metrics: ["accuracy"],
});
mis datos de entrenamiento y prueba
const yTrain = tf.tensor2d(
[10, 130, 60, 20, 150, 110, 3, 160, 99],
[1, 9]
);
const xTrain = tf.tensor2d([0, 1, 0, 0, 1, 1, 0, 1, 0], [1, 9]);
const yTest = tf.tensor2d(
[5, 106, 33, 88, 104, 140, 7, 60, 154],
[1, 9]
);
const xTest = tf.tensor2d([0, 1, 0, 0, 1, 1, 0, 0, 1], [1, 9]);
Respuestas
edkeveked
InputShape es de tamaño 2, por lo tanto, las características (aquí xtrain y xtest) deben ser de dimensión 3.
Además, no tiene sentido tener un tamaño de dimensión igual a 0 (eso significará que el tensor está vacío).
Dada su forma xtrain y xtest [a, b]
, inputShape, debería ser [b]
.
Este desajuste de forma entre el modelo y los datos de entrenamiento se ha discutido aquí y allá