【发布时间】:2018-07-13 13:52:12
【问题描述】:
总的来说,我对 Tensorflowjs 和 Tensorflow 很陌生。我有一些数据,它是 100% 使用的容量,所以一个介于 0 和 100 之间的数字,并且每天有 5 个小时记录这些容量。所以我有一个 5 天的矩阵,包含 100% 中的 5 个百分比。
我有以下型号:
const model = tf.sequential();
model.add(tf.layers.dense({units: 1, inputShape: [5, 5] }));
model.compile({ loss: 'binaryCrossentropy', optimizer: 'sgd' });
// Input data
// Array of days, and their capacity used out of
// 100% for 5 hour period
const xs = tf.tensor([
[11, 23, 34, 45, 96],
[12, 23, 43, 56, 23],
[12, 23, 56, 67, 56],
[13, 34, 56, 45, 67],
[12, 23, 54, 56, 78]
]);
// Labels
const ys = tf.tensor([[1], [2], [3], [4], [5]]);
// Train the model using the data.
model.fit(xs, ys).then(() => {
model.predict(tf.tensor(5)).print();
}).catch((e) => {
console.log(e.message);
});
我收到一个错误返回:Error when checking input: expected dense_Dense1_input to have 3 dimension(s). but got array with shape 5,5。所以我怀疑我以某种方式错误地输入或映射了我的数据。
【问题讨论】:
标签: javascript tensorflow tensorflow.js