保存流水线逻辑回归模型

Question

lrmodel=logreg_pipeline.fit(X_train_resh,y_train_resh)

lrmodel.write().overwrite().save("E:/strokestuff/strokelrpred")

lrmodel.save("E:/strokestuff/strokelrpred")

lrmodel 是一个管道，我想保存它，我的目的是保存这个模型然后加载它部署在Flutter 中。我已经尝试了所有的解决方案，有人可以帮助我吗？

Answer 1

您可以使用 joblib 将您的模型保存在 .joblib 文件中：

import joblib

pipe_clf_params = {}
filename = 'E:/strokestuff/strokelrpred/strokelrpred.joblib'

pipe_clf_params['pipeline'] = lrmodel
joblib.dump(pipe_clf_params, filename)

保存流水线逻辑回归模型

Save a pipeline Logistic Regression model

flutter

machine-learning

pipeline