【发布时间】:2019-12-09 08:22:36
【问题描述】:
我正在尝试运行代码
import data_processing as dp
import numpy as np
test_set = dp.read_data("./data2019-12-01.csv")
import tensorflow as tf
import keras
def train_model():
autoencoder = keras.Sequential([
keras.layers.Flatten(input_shape=[400]),
keras.layers.Dense(150,name='bottleneck'),
keras.layers.Dense(400,activation='sigmoid')
])
autoencoder.compile(optimizer='adam',loss='mse')
return autoencoder
trained_model=train_model()
trained_model.load_weights('./weightsfile.h5')
trained_model.evaluate(test_set,test_set)
第 3 行中的 test_set 是形状为 (3280977,400) 的 numpy 数组。我正在使用 keras 2.1.4 和 tensorflow 1.5。
但是,这会引发以下错误
ValueError: Input 0 is in compatible with layer flatten_1: expected min_ndim=3, found ndim=2
我该如何解决?我尝试更改 flatten 层中的 input_shape 并在互联网上搜索可能的解决方案,但没有一个解决。有谁可以帮我离开这里吗?谢谢
【问题讨论】:
标签: machine-learning deep-learning flatten