在完成机器学习项目后,如何保存它供其他人查看

时间:2018-10-14 08:10:13

标签: python machine-learning

在与您的机器学习项目一起工作后,我需要您的帮助才能成为机器学习的初学者。如何将其发送给其他人看,请帮助使用Python

1 个答案:

答案 0 :(得分:1)

pickle是在Python中序列化对象的标准方法。

假设您有一些数据,进行了一些预处理并拟合了模型。

model = LogisticRegression()
model.fit(X_train, Y_train)

您可以使用pickle将那个模型对象作为二进制文件保存到磁盘上。

import pickle
filename = 'finalized_model.sav'
pickle.dump(model, open(filename, 'wb'))

当您想再次使用它时,可以像这样重新加载模型,

loaded_model = pickle.load(open(filename, 'rb'))

然后再次使用...

result = loaded_model.score(X_test, Y_test)
print(result)