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

问题描述 投票:0回答:1

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

machine-learning deep-learning neural-network saving-data
1个回答
0
投票

如果使用的是python,则可以使用腌制另存为对象

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)
© www.soinside.com 2019 - 2024. All rights reserved.