Новичок: непонимание формы [, 1] на Tensorflow.js - PullRequest
Новая
       27

Новичок: непонимание формы [, 1] на Tensorflow.js

1 голос
/ 27 мая 2019

Я хочу создать модель для преобразования Цельсия в градусы Фаренгейта с помощью Tensorflow.js (с Node.js).

Однако я не понимаю, какие формы использовать.

Я пробовал другие input_shape, такие как [1], [1,20] и, наконец, установил его на[20] Я также пробовал различную тензорную форму для массивов Цельсия и Фаренгейта, таких как tensor(celsius), tensor([celsius]).

Вот код 


var model = tf.sequential()
model.add(tf.layers.dense({inputShape:[20], units: 1}))

async function trainModel(model, inputs, labels) {
    // Prepare the model for training.  
    model.compile({
      optimizer: tf.train.adam(),
      loss: tf.losses.meanSquaredError,
      metrics: ['mse'],
    });

    const batchSize = 28;
    const epochs = 500;

    return await model.fit(inputs, labels, {
      epochs,
      shuffle: true,
    //   callbacks: tfvis.show.fitCallbacks(
    //     { name: 'Training Performance' },
    //     ['loss', 'mse'], 
    //     { height: 200, callbacks: ['onEpochEnd'] }
    //   )
    });
  }

c = tf.tensor([celsius]) // celsius = [1,2,3,4,...]
console.log(c.shape) // ==> [1,20]

f = tf.tensor([fahrenheit])
console.log(f.shape) // ==> [1,20]

trainModel(model, c, f)

Более того, в Pythonучебник input_shape это [1].С Node.js, кажется, работает только [20].

Форма входов [1,20], и это правильно.

Форма меток также [1,20], но вызывает следующую ошибку:

Отладчик говорит: 

Error when checking target: expected dense_Dense1 to have shape [,1], but got array with shape [1,20].

- РЕДАКТИРОВАТЬ

Кроме того, когда япопробуйте input_shape: [1,20], это даст мне: 

expected dense_Dense1_input to have 3 dimension(s). but got array with shape 1,20

-

Я ожидаю, что модель будет тренироваться, связав значения C ° со значениями F °.

Спасибо

1 Ответ

1 голос
/ 27 мая 2019

Ошибка ясна: 

{inputShape:[20], units: 1}

Модель содержит один слой. inputShape:[20], что означает, что batchInputShape, равное [null, 20], будет формой первого слоя. Аналогично, units: 1 указывает, что последний слой будет иметь форму [null, 1].

Используемые элементы имеют форму [1, 20], соответствующую, следовательно, batchInputShape модели. Однако это не относится к этикеткам, имеющим форму [1, 20]. Он должен иметь форму [1, 1], поэтому выдается ошибка:

ожидал, что dens_Dense1 будет иметь форму [, 1], но получил массив с формой [1,20]

Размер единиц измерения модели необходимо изменить, чтобы отразить форму меток. 

{inputShape:[20], units: 20}
...