如何将训练好的模型移动到另一个Python笔记本？

Question

我有一个Python笔记本，其中包含多个函数，例如pre_process等，并且它包含我训练的模型。现在我应该提交一个 python 笔记本，其中包含我训练过的模型以进行进一步测试。我怎样才能做到这一点？

我知道在运行单元格时，它会保存所有单元格使用的变量，我可以链接这两个文件并从那里调用模型吗？

Answer 1

您通常希望将机器学习保存为 Pickle (.pkl) 文件。

# Saving the model
import pickle
with open('model.pkl', 'wb') as file:
    pickle.dump(model, file)

稍后可以再次加载。

with open('model.pkl', 'rb') as file:
    loaded_model = pickle.load(file)

还有其他方法，具体取决于您如何训练模型。