修正model.fit_generator

原程式碼:model.fit_generator(datagen.flow(x_train,y_train,batch_size=64),steps_per_epoch = 1000,epochs =10,validation_data=(x_test,y_test),workers=4,verbose=1)

C:\Users\User\AppData\Local\Temp\ipykernel_312\2215498590.py:1: UserWarning: `Model.fit_generator` is deprecated and will be removed in a future version. Please use `Model.fit`, which supports generators.

修正成這樣就好了
model.fit(datagen.flow(x_train, y_train, batch_size=64), 
          steps_per_epoch=1000,
          epochs=10,
          validation_data=(x_test, y_test),
          workers=4,
          verbose=1)

评论

此博客中的热门博文

修正input_img_data = np.random.random((1, 150, 150, 3)) * 20 + 128.

緣起