我正在研究机器学习项目。我已经在我的数据上应用截断的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 pickle库
import pickle
# To save
pickle.dump(model, "model.p")
# To load again
with open('model.p', 'r') as fp:
model = pickle.load(fp)