Документация Engee

Сохранение моделей на диск

Использование JLD.jl является предпочтительным способом сохранения ScikitLearn.jl моделей. Если вы также используете модели Python (с помощью @sk_import), вам придется импортировать и PyCallJLD.

using PyCall, JLD, PyCallJLD
using ScikitLearn
using ScikitLearn.Pipelines
@sk_import decomposition: PCA
@sk_import linear_model: LinearRegression

pca = PCA()
lm = LinearRegression()

X=rand(10, 3); y=rand(10);

pip = Pipeline([("PCA", pca), ("LinearRegression", lm)])
fit!(pip, X, y)   # соответствие некоторому набору данных

JLD.save("pipeline.jld", "pip", pip)
pip = JLD.load("pipeline.jld", "pip") # Загрузка конвейера обратно