如何使用经过训练的模型绘制精度

Question

我有一个Tensorflow模型已经在我的笔记本上训练过了，我想在那之后绘制准确度和损失

这是我的密码：

myGene = trainGenerator(2,'/content/data/membrane/train','image','label',
                       data_gen_args,save_to_dir = None)
model = unet()
model_checkpoint = ModelCheckpoint('unet_membrane.hdf5', 
                             monitor='loss',verbose=1, save_best_only=True)
model.fit_generator(myGene,steps_per_epoch=2000,
                    epochs=5,callbacks=[model_checkpoint])

有没有办法策划什么？ 因为我试过使用matplotlib，但它不起作用

import matplotlib.pyplot as plt

plt.plot(history['accuracy'])
plt.plot(history['loss'])

Answer 1

试试这个：

history = model.fit_generator(myGene,
              steps_per_epoch=2000,
              epochs=5,callbacks=[model_checkpoint])

然后，对于绘图：

plt.plot(history.history['accuracy'])
plt.plot(history.history['loss'])