机器学习_模型的保存和加载_模块导入错误: ImportError: cannot import name ‘joblib‘ from ‘sklearn.externals‘
1.如果想要让模型训练的数据得到的结果一致, 需要保证模型使用的训练集和测试集是一样的, 可以设置一下算法的random_state来确保一致。从scikit-learn 0.24 版本开始, joblib 已经不再包含在 sklearn.externals 子模块中了。3.加载模型需要一个变量进行承接, 为了方便, 可以用estimator来承接。filename-要加载读取的模块文件名和路径。
原本不想写这篇文章的, 内容太少, 但是有一个地方要注意,
sklearn自从0.24版开始, 就把joblib模块移除了, 所以使用较新版本sklearn模块时, 是无法直接导
入joblib模块的, 需要在终端安装joblib, 然后再使用
如果硬要通过sklearn导入, 就会报错咯, 像这种
ImportError: cannot import name 'joblib' from 'sklearn.externals' (D:\developer\python3.12.3(64-bit)\Lib\site-packages\sklearn\externals\__init__.py)

模型的保存和加载
api:
需要先安装模块joblib,然后直接导入即可, 步骤如下
1.pip install joblib
2.import joblib
因为在较新版本的 scikit-learn 中,joblib模块已经被移除了
从scikit-learn 0.24 版本开始, joblib 已经不再包含在 sklearn.externals 子模块中了
保存:
joblib.dump()
参数:
vlaue-训练出来的模型对象
value=estimator
filename-保存文件的名字和路径
'./data/model_name.pkl'
下载:
joblib.load()
参数:
filename-要加载读取的模块文件名和路径
'./data/model_name.pkl'
注意:
1.如果想要让模型训练的数据得到的结果一致, 需要保证模型使用的训练集和测试集是一样的, 可以设置一下算法的random_state来确保一致
2.保存文件, 文件的后缀名**.pkl
3.加载模型需要一个变量进行承接, 为了方便, 可以用estimator来承接
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)