简单的问题和我确定的答案很简单,但我真的很难将模型形状与适合模型的张量相匹配。
这个简单的代码
let tf = require('@tensorflow/tfjs-node');
let features = {
x: [1,2,3,4,5,6,7,8,9],
y: [1,2,3,4,5,6,7,8,9]
}
let tensorfeature = tf.tensor2d(Object.values(features))
console.log(tensorfeature.shape)
const model = tf.sequential();
model.add(tf.layers.dense(
{
inputShape: tensorfeature.shape,
units: 1
}
))
const optimizer = tf.train.sgd(0.005);
model.compile({optimizer: optimizer, loss: 'meanAbsoluteError'});
model.fit(tensorfeature,
{epochs: 5}
)
导致错误:检查输入时出错:预期dense_Dense1_input具有3个维度。但得到了形状为 2,9 的数组
尝试了多种重塑、切片等方法,但没有成功。有人能指出我到底出了什么问题吗?