【发布时间】:2020-06-18 09:45:13
【问题描述】:
简单的问题,我肯定答案很简单,但我真的很难将模型形状与张量拟合到模型中。
这个简单的代码
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 的数组
用 reshape、slice 等尝试了多种方法,但都没有成功。有人能指出我到底出了什么问题吗?
【问题讨论】: