sklearn 存取模型

来源:互联网 发布:博雅软件股份有限公司 编辑:程序博客网 时间:2024/05/16 09:00

整理模型存取方法

  • 路径存在判断
import osif os.path.exists("./model/"):    #判断路径是否存在
  • 路径创建
if not os.path.exists("./model/"):    os.mkdir("./model/")
  • 模型下载
from sklearn.externals import joblibimport osdef download_model(self,classification):    if not os.path.exists("./model/"):        os.mkdir("./model/")    joblib.dump(classification, "./model/"+"train_model.m")
  • 模型加载
from sklearn.externals import joblibimport osdef load_model(self,groupnum):    classification = joblib.load("./model/"+"train_model.m")    return classification
原创粉丝点击