Блог

Я создал свою первую модель, но прогнозы составлены не в том формате. Как мне удалить измерение в выходных данных моего прогноза (или изменить мой последний слой, чтобы получить правильный)?

 const actualYs = [1,2,3] // The shape of my values Y

const predictions = [[1],[2],[3]] // The shape of my predictions

// My last layer looks like this:
model.add(tf.layers.dense({ units: 1, useBias: true }))
 

Так что, исходя из моего ограниченного понимания. Может быть, я мог бы удалить измерение predictions или изменить последний слой? Но я уже поставил 1 , так что не уверен, на что еще я мог бы его установить.

В случае, если это поможет, это мой фактический console.log

 MY Y VALUES

Tensor
    [0.0862738, 0.0862553, 0.0861815, ..., 0.0054516, 0.0043004, 0.0037461]

PREDICTIONS

Tensor
    [[0.1690691],
     [0.1659686],
     [0.1698797],
     ...,
     [0.1118171],
     [0.1092742],
     [0.1096415]]
 

Я хочу, чтобы прогнозы выглядели как мои фактические значения Y.

Заранее спасибо.

reshape или squeeze может быть использован

 const x = tf.tensor([[1],[2],[3]] ).reshape([-1]);
// or 
const x = tf.tensor([[1],[2],[3]] ).squeeze();