在Jupyter Notebook中运行经过训练的Kaggle模型

Question

我已经在kaggle中构建并训练了模型，并下载了其输出。现在如何在Jupyter Notebook中本地运行经过训练的模型以进行预测？

Answer 1

您可以使用以下方式将模型保存到文件中：

model.save('model.hdf5')

然后您可以在kaggle内核中运行它：

from IPython.display import FileLink
FileLink(r'model.h5')

然后它将生成一个链接，以便您可以下载hdf5文件。

在本地jupyter笔记本中运行以下命令：

from keras.models import load_model
model = load_model('model.h5')

Answer 2

我猜您正在尝试说如何在jupyter中加载预训练的模型权重。

您要做的就是在jupyter中创建一个新模型，与kaggle中相同的预训练模型层和参数一样。然后使用loadweights方法。您现在就可以预测现在的欢呼了：D

例如：

weights_path="donwloaded_weights.hdf5"
model = YourModel()
model.loadweights(weights_path)
model.predict(your_data)