I Raun Этот код, и я получил ошибку ниже: < /p>
Ошибка: ошибка при проверке цели: ожидаемый dense_dense2, чтобы иметь 2 размерных (ах). Но у меня есть массив с формой 2,5,2 < /p>
< /blockquote>
Кто-нибудь может мне помочь?const tf = require('@tensorflow/tfjs');
let log = console.log;
let value_x = [
[1, 1, 1, 1, 1],
[1, 1, 1, 1, 0]
];
let value_y = [
[
[1, 0],
[1, 0],
[1, 0],
[1, 0],
[1, 0]
],
[
[1, 0],
[1, 0],
[1, 0],
[1, 0],
[1, 0]
]
];
const model = tf.sequential();
model.add(tf.layers.dense({
inputShape: [5],
units: 10
}));
model.add(tf.layers.dense({
units: 5
}));
model.compile({
loss: 'meanAbsoluteError',
optimizer: 'sgd',
metrics: ['mse']
}); //adam
const xs = tf.tensor2d(value_x, [2, 5]);
const ys = tf.tensor3d(value_y, [2, 5, 2]);
xs.print();
ys.print();
log();
model.fit(xs, ys, {
epochs: 10,
callbacks: {
onEpochEnd: (epoch, log) => {
console.log(epoch, '- ', log.loss);
}
}
}).then(() => {
const a = model.predict(tf.tensor2d([1, 1, 1, 0, 0], [1, 5])).arraySync();
log(a);
});
Подробнее здесь: https://stackoverflow.com/questions/796 ... i-go-wrong
