如何在 python 中保存截断的 svd 模型答案

【问题标题】：How to save a trunckated svd model in python如何在 python 中保存截断的 svd 模型
【发布时间】：2019-03-13 18:54:18
【问题描述】：

我正在做一个机器学习项目。我在我的数据上应用了截断的 svd 以减少特征，然后在该数据上训练神经网络。我已经使用 to_json() 保存了神经网络模型。如何保存截断的 svd 模型，以便以后转换看不见的数据。

这是我使用截断 svd 的方法。

from sklearn.decomposition import TruncatedSVD
model = TruncatedSVD(n_components=600,n_iter=10).fit(train_features)
train_features= model.transform(train_features)
test_features= model.transform(test_features)`

我希望我把这个问题说清楚了。有人可以帮我解决这个问题吗？

【问题讨论】：

标签： python machine-learning pca svd truncated

【解决方案1】：

你可以使用 python pickle 库

import pickle
# To save
pickle.dump(model, "model.p")
# To load again
with open('model.p', 'r') as fp:
    model = pickle.load(fp)

【讨论】：

我们需要为此创建一个文件对象：file_obj = open("model.p", "wb") ; pickle.dump(model, file_obj)