保存自己训练的分类模型
2020-04-15 本文已影响0人
一只不会南飞的燕
今天要完成分类模型的web端可视化,首先我要保存用训练数据提前训练好的分类模型。那么使用python语言怎么保存在即训练的分类模型呢?
from sklearn.externals import joblib
#保存分类模型
joblib.dump(model, "train_model.m")
#从本地调回分类模型
#model = joblib.load("train_model.m")
我是一只不会南飞的燕!大家一起学习,一起进步!