Форма ввода и керас

Question

У меня есть набор обучающих данных, где каждый вход представляет собой вектор длиной 138. У меня есть 519 из этих векторов для размера партии 519. Это не изображения, а просто действительные числа.

Я пытаюсь начать с двухслойной плотной модели Keras:

model = keras.Sequential([
    layers.Dense(200, activation=tf.nn.relu, input_shape=[138]),
    layers.Dense(200, activation=tf.nn.relu),
    layers.Dense(1)
])

Когда я строю модель, я получаю следующую ошибку:

Error when checking input: expected dense_27_input to have shape (138,) but got array with shape (519,).

Где в Keras я различаю партиюразмер из числа входных функций?layers.Dense(), кажется, предполагает, что мой ввод в строках против столбцов.

Answer 1

Керас ожидает, что первая ось будет осью пакета.Поэтому, если у вас есть 519 обучающих выборок, каждый из которых представляет собой вектор длиной 138, массив, который вы передаете методу fit, должен иметь форму (519, 138).Поэтому, если в настоящее время массив обучающих данных имеет форму (138, 519), просто транспонируйте его, чтобы сделать форму согласованной:

import numpy as np

train_data = np.transpose(train_data)