lrmodel=logreg_pipeline.fit(X_train_resh,y_train_resh)
lrmodel.write().overwrite().save("E:/strokestuff/strokelrpred")
lrmodel.save("E:/strokestuff/strokelrpred")
lrmodel 是一个管道模型,我想保存它,我的目标是保存这个模型,然后在 Flutter 中加载并部署它。我已经尝试了所有找到的解决方案,有人能帮我吗?
回答:
您可以使用 joblib
将模型保存为 .joblib
文件:
import joblibpipe_clf_params = {}filename = 'E:/strokestuff/strokelrpred/strokelrpred.joblib'pipe_clf_params['pipeline'] = lrmodeljoblib.dump(pipe_clf_params, filename)