如何在电脑上保存机器学习模型集合?

Question

我正在使用Google Colab，我的运行时一直断开连接。与其一次又一次地训练它，我想我会以某种方式将其存储在 PC 中。因此，如果我必须演示它，我可以加载该文件并使用 .predict。 我需要将其保存到我的 PC，以便在我正在使用的平台上出现问题时，我可以使用简单的 .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)