javascript - Tensorflowjs 在训练时产生形状错误

Question

我正在尝试使用 tensorflow 版本 tfjs@0.12.0 训练一个简单的网络模型，并且我正在使用图层 API。它只是一个将两个数字作为输入并返回相同的两个数字的网络 - 因此网络应该尝试学习单位矩阵。

我正在使用 6 个示例进行训练，因此我的输入和输出的形状都是 [6,2]，但是我得到了一个形状错误。

我的javascript文件如下:

function setup() {
    const model = tf.sequential();

    const hidden = tf.layers.dense({
        units: 4,
        useBias: true,
        activation:'sigmoid',
        inputDim: [2],
    });

    model.add(hidden);

    const output = tf.layers.dense({
        units: 2,
        activation:'sigmoid',
    });

    model.add(output);

    model.compile({
        optimizer: 'sgd',
        loss: 'meanSquaredError',
    });

    const xs = tf.tensor2d([
        [0.25, 0.92],
        [0.12,0.3],
        [0.4,0.74],
        [0.3,0.82],
        [0.09,0.95],
        [0.53,0.2],
    ]);

    const ys = tf.tensor2d([
        [0.25, 0.92],
        [0.12,0.3],
        [0.4,0.74],
        [0.3,0.82],
        [0.09,0.95],
        [0.53,0.2],
    ]);


    console.log(model.fit(xs,ys));
}

但是当我运行脚本时出现形状错误:

Promise { <state>: "rejected" }
sketch.js:46:5
Error: Error when checking input: expected dense_Dense1_input to have shape [,2], but got array with shape [6,2].

Answer 1

在您的隐藏 层声明中设置inputDim: 2 或inputShape: [2]。