machine-learning - 如何在我的 PC 上保存机器学习模型集合？

Question

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

Answer 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)