Как сохранить транкированную модель SVD в Python

Question

Я работаю над проектом машинного обучения.Я применил усеченный 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)`

Я надеюсь, что разъяснил этот вопрос.Может ли кто-нибудь помочь мне решить этот вопрос?

Answer 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)