原本不想写这篇文章的, 内容太少, 但是有一个地方要注意,

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来承接

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐