【发布时间】:2021-03-31 05:33:25
【问题描述】:
我尝试拟合我的模型,但遇到“类型错误”。这段代码的编写与 kaggle CNN 源代码示例中描述的几乎相同。
https://www.kaggle.com/kanncaa1/convolutional-neural-network-cnn-tutorial/notebook
但是,不断出现类型错误。我将model.fit_generator 更改为model. fit,因为该函数是从新版本的tensorflow 更新而来的。我猜想shape[0] 可以解决这个与内部相关的问题。谁能帮忙指出下面代码中的错误类型?
# model fitting
# from tensorflow 2.1.0
history = model.fit(datagen.flow(train_x, train_y, batch_size = batch_size),
epochs = epochs,
validation_data = (valid_x, valid_y),
steps_per_epoch = train_x.shape[0] // batch_size)
【问题讨论】:
-
你最好检查一下 train_x,好像 train_x.shape 返回一个整数。您可能在加载数据时遇到问题,因为您的输入形状只有一个暗淡。
标签: python tensorflow deep-learning conv-neural-network