模型保存和加载

水深无声 2022-04-23 03:22 333阅读 0赞

我们可以将训练后的模型保存下来，下次直接导出就行了，节省了时间。

### 代码： ###

from sklearn.datasets import load_boston
    from sklearn.model_selection import train_test_split
    from sklearn.preprocessing import StandardScaler
    from sklearn.linear_model import Ridge
    from sklearn.metrics import mean_squared_error
    from sklearn.externals import joblib
    
    def linear3():
        """
        岭回归对波士顿房价进行预测
        :return:
        """
        # 1）获取数据
        boston = load_boston()
        print("特征数量：\n", boston.data.shape)
    
        # 2）划分数据集
        x_train, x_test, y_train, y_test = train_test_split(boston.data, boston.target, random_state=22)
    
        # 3）标准化
        transfer = StandardScaler()
        x_train = transfer.fit_transform(x_train)
        x_test = transfer.transform(x_test)
    
        # 4）预估器
        #estimator = Ridge(alpha=0.5, max_iter=10000)
        #estimator.fit(x_train, y_train)
    
        # 保存模型
        #joblib.dump(estimator, "my_ridge.pkl")
        # 加载模型
        estimator = joblib.load("my_ridge.pkl")
    
        # 5）得出模型
        print("岭回归-权重系数为：\n", estimator.coef_)
        print("岭回归-偏置为：\n", estimator.intercept_)
    
        # 6）模型评估
        y_predict = estimator.predict(x_test)
        print("预测房价：\n", y_predict)
        error = mean_squared_error(y_test, y_predict)
        print("岭回归-均方误差为：\n", error)
    
        return None
    
    if __name__ == "__main__":
        linear3()

发表评论取消回复

表情：

评论列表（有 0 条评论，333人围观）

还没有评论，来说两句吧...

相关阅读

相关 PyTorch 06： PyTorch保存和加载模型

下面学习如何使用 PyTorch 保存和加载模型。我们经常需要加载之前训练过的模型，或继续用新的数据训练模型。所以这部分还是挺重要的。 %matplotlib inl

梦里梦外;/ 2023年03月13日 15:29/ 0 赞/ 20 阅读

相关 pytorch保存模型pth_PyTorch之保存加载模型

前提 SAVING AND LOADING MODELS 当提到保存和加载模型时，有三个核心功能需要熟悉： 1.torch.save：将序列化的对象保存到disk。这个函

Bertha 。/ 2022年12月31日 15:30/ 0 赞/ 315 阅读

相关 sklearn模型保存与加载

sklearn模型保存与加载 sklearn模型的保存和加载API 线性回归的模型保存加载案例保存模型 sklearn模型的保存和加载API

我会带着你远行/ 2022年12月29日 14:00/ 0 赞/ 259 阅读

相关 Pytorch之模型加载/保存

pytorch保存模型有两种方法： 1. 保存整个模型 (结构+参数) 2. 只保存参数（官方推荐）两者都是用`torch.save(obj, dir)`实现，这个函

た入场券/ 2022年11月27日 00:51/ 0 赞/ 290 阅读

相关 Tensorflow模型的保存和加载(一)

刚接触深度学习，Tensorflow模型的保存和加载尚不清楚，根据教程的翻译做一记录，不当之处敬请指正。原文地址：[http://cv-tricks.com/tensorf

约定不等于承诺〃/ 2022年11月25日 13:17/ 0 赞/ 251 阅读

相关保存和加载模型

本文档提供了有关 PyTorch 模型保存和加载的各种用例的解决方案。随意阅读整个文档，或者直接跳到所需用例所需的代码。在保存和加载模型时，需要熟悉三个核心功能： 1.

刺骨的言语ヽ痛彻心扉/ 2022年09月11日 00:29/ 0 赞/ 273 阅读

相关 Tensorflow模型保存与加载

Tensorflow模型保存与加载 import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

小咪咪/ 2022年09月08日 00:09/ 0 赞/ 375 阅读

相关 Tensorflow模型保存和加载

翻译自：http://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-comple

蔚落/ 2022年05月29日 04:44/ 0 赞/ 413 阅读

相关模型保存和加载

我们可以将训练后的模型保存下来，下次直接导出就行了，节省了时间。代码： from sklearn.datasets import load_boston

水深无声/ 2022年04月23日 03:22/ 0 赞/ 334 阅读

相关 python保存和加载机器学习模型

一：使用pickle实现 Save Model Using Pickle import pandas from sklearn import mod

素颜马尾好姑娘i/ 2022年02月09日 14:21/ 0 赞/ 351 阅读