查看历史记录时，“NoneType”对象不可下标

Question

我想加载一个模型并将其显示在绘图中。不幸的是，我收到了一个错误。

为什么此时会出现这个错误（通常他应该知道？），我该如何解决？

我已经调查过这个错误并寻找类似Python Math - TypeError: 'NoneType' object is not subscriptable的问题

为什么history.history['loss'] 会出现此错误？通常他应该知道！

错误：

training_loss = history.history['loss']
TypeError: 'NoneType' object is not subscriptable

代码：

def load_model():
    history = tf.keras.models.load_model(path)
    return model

def get_loss(history):
    # Get training and test loss histories
    training_loss = history.history['loss'] # here is the error
    test_loss = history.history['val_loss']

    # Create count of the number of epochs
    epoch_count = range(1, len(training_loss) + 1)

    # Visualize loss history
    plt.plot(epoch_count, training_loss, 'r--')
    plt.plot(epoch_count, test_loss, 'b-')
    plt.legend(['Training Loss', 'Test Loss'])
    plt.xlabel('Epoch')
    plt.ylabel('Loss')
    plt.show();

get_loss(load_model)

---

编辑：

history = model.fit(...)
model.save(model_file, overwrite=True)

Answer 1

tf.keras.models.load_model

是一种加载模型的方法。

模型不是历史。

在张量流中。定义模型后。你训练它：

model = your defination
history = model.fit(**kwargs)

然后，您应该保存模型和历史记录。并加载历史来绘制图片。

这是：

model.save(your path)
joblib.dump(history, your history path)

然后你应该加载你的历史记录：

history = joblib.load(your history path)