Массив тензоров, который вы передаете своей модели, не соответствует ожидаемому размеру. Ожидается увидеть 1 Тензор (ы) в тензорном потоке js - PullRequest
Новая
       16

Массив тензоров, который вы передаете своей модели, не соответствует ожидаемому размеру. Ожидается увидеть 1 Тензор (ы) в тензорном потоке js

0 голосов
/ 07 апреля 2020

Я делаю модель с размерами 145, 153, но вы продолжаете получать ошибку, что форма входа не такая, как ожидалось. не могли бы вы помочь, что я делаю неправильно. код ниже 

const model = tf.sequential();

model.add(tf.layers.dense({
    units: 128,
    inputShape: [153, ],
    activation: 'relu'
}))

model.add(tf.layers.dropout({

    rate: 0.5

}))
model.add(tf.layers.dense({
    units: 64,
    activation: 'relu'
}))
model.add(tf.layers.dropout({

    rate: 0.5

}))
model.add(tf.layers.dense({

    units: o,
    activation: 'softmax'

}))
model.compile({
    loss: 'categoricalCrossentropy',
    optimizer: 'sgd',
    matrics: ['accuracy']

})

const startTime = Date.now();
model.fit(training, output, {
        epoch: 200,
        batchSize: 5,
        verbose: 1
    })
    .then((history) => {
        console.log("model created  ", Date.now() - startTime)
    });
...