在keras中设置模型的输入形状

时间:2018-06-12 17:34:59

标签: python input keras shape

我在张量流上看到了其他类似的问题,但与我的问题不符。

型号:

# picture size
img_row = 128
img_col = 647
shape = (img_row, img_col)

img = Input(input_shape)
...

结果 model

数据:

有1000个数据,每个数据都有形状(128,647),还有一列Dataframe df。 因此,大小结果和数据预览如下: enter image description here enter image description here

问题

问题是:当我将数据传递给模型时,发生了一些大小错误。

train_history = model.fit(  x = df["data"],
                            y = df["genre_idx"],
                            validation_split = 0.1,
                            epochs = 30,
                            batch_size = 200,
                            verbose = 2
                         )

错误信息如下:

Error when checking input: expected input_79 to have 3 dimensions, but got array with shape (1000, 1)

这可能是一个很低的问题,但我没有弄清楚这种情况的主要问题是什么以及如何解决它。

1 个答案:

答案 0 :(得分:0)

您需要将其作为单个 ndarray 提供,您可以使用数据框的.values property进行提取。输入的预期形状为(1000, 128, 647)