首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >fit_generator()缺少1个必需的位置参数:“generator”

fit_generator()缺少1个必需的位置参数:“generator”
EN

Stack Overflow用户
提问于 2021-07-08 22:14:27
回答 2查看 70关注 0票数 0

需要帮助才能弄清楚这一点。我不确定哪里出了问题,但错误仍然存在。环顾四周,但找不到类似的问题。

代码语言:javascript
复制
import matplotlib.pyplot as plt
from PIL import Image
import os
import numpy as np
from skimage import io
from keras.preprocessing.image import ImageDataGenerator
from matplotlib import cm
from mpl_toolkits.axes_grid1 import ImageGrid
import math
%matplotlib inline
import keras
import tensorflow as tf
from keras.models import Model

batch_size=32
datagen_args = dict(rotation_range=20,
    width_shift_range=0.2,
    height_shift_range=0.2,
                   rescale=1./255)
datagen = ImageDataGenerator(**datagen_args)
train_datagenerator = datagen.flow_from_directory('/content/drive/MyDrive/cats_dogs_small/train',target_size=(128,128),
                                            batch_size=batch_size,interpolation="lanczos",shuffle=True)
valid_datagenerator = datagen.flow_from_directory('/content/drive/MyDrive/cats_dogs_small/validation',target_size=(128,128),
                                            batch_size=batch_size,interpolation="lanczos",shuffle=True)

epochs = 25
hist = Model.fit_generator(train_datagenerator, 
                           steps_per_epoch= math.ceil(train_datagenerator.samples//batch_size),
                           epochs=epochs, validation_data=valid_datagenerator, validation_steps=math.ceil(valid_datagenerator.samples//batch_size),verbose = 1, workers=8)

错误消息如下:

代码语言:javascript
复制
TypeError                                 Traceback (most recent call last)
<ipython-input-69-178574fd407f> in <module>()
      2 hist = Model.fit_generator(train_datagenerator, 
      3                            steps_per_epoch= math.ceil(train_datagenerator.samples//batch_size),
----> 4                            epochs=epochs, validation_data=valid_datagenerator, validation_steps=math.ceil(valid_datagenerator.samples//batch_size),verbose = 1, workers=8)

TypeError: fit_generator() missing 1 required positional argument: 'generator'
EN

回答 2

Stack Overflow用户

发布于 2021-07-09 14:54:02

fit生成器已折旧,只需使用model.fit即可。注意,您使用的是Model.fit_generator。您应该使用model.fit。

票数 1
EN

Stack Overflow用户

发布于 2021-07-10 15:13:38

我意识到我没有定义什么是模型。添加了层,编译并再次尝试这次它可以工作。我是新手,还有很多东西要学!

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68303356

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档