如何在我的 PC 上保存机器学习模型集合？答案

【问题标题】：How do I save an Machine Learning ensemble of models on my PC?如何在我的 PC 上保存机器学习模型集合？
【发布时间】：2020-01-17 18:15:36
【问题描述】：

我正在使用 Google Colab，但我的运行时一直断开连接。与其一次又一次地训练它，我想我会以某种方式将它存储在 PC 中。因此，如果我必须演示它，我可以加载该文件并使用 .predict。我需要将它保存到我的 PC 上，以便在我正在使用的平台上断开连接时，我可以使用简单的 .load() 函数运行它。

【问题讨论】：

你没有提到你的工具，没有展示你的代码，也没有谈论你尝试过的东西。请更新您的问题，以便我们知道如何最好地回答它。
@erip 我的问题很笼统；唯一的特殊性在于我使用的是 Colab。问题已得到解答。

标签： machine-learning deep-learning neural-network saving-data

【解决方案1】：

如果你使用的是 python，你可以使用 pickling 来保存为一个对象

import pickle
model = LogisticRegression()
model.fit(X_train, Y_train)
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))
# load the model from disk
loaded_model = pickle.load(open(filename, 'rb'))
result = loaded_model.score(X_test, Y_test)
print(result)

【讨论】：

它会保存到我的电脑吗？您能否详细说明 .dumps() 行中发生了什么？
它将您的训练模型转换为对象并保存在您的计算机上。（阅读有关序列化和反序列化或（酸洗和解酸）的信息）你会明白的。