当前位置: 首页 > news >正文

SVM 保存和加载模型参数

在 Python 中,你可以使用 scikit-learn 库中的 joblib 或 pickle 模块来保存和加载 SVM 模型的参数。以下是一个简单的示例代码,演示了如何使用 joblib 模块保存和加载 SVM 模型的参数:

保存模型参数:

from sklearn import svm
from sklearn import datasets
import joblib

# 载入数据集
iris = datasets.load_iris()
X, y = iris.data, iris.target

# 创建 SVM 模型
model = svm.SVC()
model.fit(X, y)

# 保存模型参数
joblib.dump(model, 'svm_model.pkl')

加载模型参数

# 加载模型参数
loaded_model = joblib.load('svm_model.pkl')

# 使用加载的模型进行预测
result = loaded_model.predict([X[0]])
print(result)

在这个示例中,我们首先使用 scikit-learn 载入了鸢尾花数据集,并创建了一个 SVM 模型。然后,我们使用 joblib.dump 函数将模型参数保存到名为 svm_model.pkl 的文件中。接着,我们使用 joblib.load 函数加载保存的模型参数,并使用加载的模型进行预测。

你也可以使用 pickle 模块来保存和加载模型参数,示例如下:

保存模型参数:

import pickle

# 保存模型参数
with open('svm_model.pkl', 'wb') as f:
    pickle.dump(model, f)

加载模型参数:

# 加载模型参数
with open('svm_model.pkl', 'rb') as f:
    loaded_model = pickle.load(f)

# 使用加载的模型进行预测
result = loaded_model.predict([X[0]])
print(result)

这段代码使用了 pickle.dump 函数将模型参数保存到名为 svm_model.pkl 的文件中,然后使用 pickle.load 函数加载保存的模型参数,并使用加载的模型进行预测。

http://www.dtcms.com/a/6089.html

相关文章:

  • 【数据结构】八大排序之快速排序算法
  • Android设备sdcard/tf卡不识别在电脑上可以
  • 可变参数应用(C++并发编程中的joining_thread代码)
  • 解决Android AAPT: error: resource android:attr/lStar not found. 问题
  • CMake入门教程【基础篇】列表操作(list)
  • 超简单|配图详细| 双系统Ubuntu22.04 系统磁盘扩容
  • 旅游平台网页前后端
  • Selenium教程04:鼠标+键盘网页的模拟操作
  • 基于SpringBoot的物流管理系统
  • linux系统shell流文本处理工具——awk
  • 使用IDEA官方docker插件构建镜像
  • 速学python·注释
  • openssl 命令详解
  • MySQL的CRUD操作函数介绍union和union all
  • 2023年终总结
  • Elasticsearch:使用 ELSER v2 文本扩展进行语义搜索
  • RoadMap8:C++中类的封装、继承、多态与构造函数
  • 宣传照(私密)勿转发
  • 使用.Net nanoFramework为ESP32进行蓝牙配网
  • Django 文件上传(十二)
  • prometheus grafana mysql监控配置使用
  • 【JAVA】泛型和Object的区别
  • 深入了解Swagger注解:@ApiModel和@ApiModelProperty实用指南
  • 外包干了3个月,技术退步明显.......
  • 使用 dbgate 在 sealos 上完美管理 mysql pgsql 等数据库
  • 14:00面试,14:08就出来了,问的问题过于变态了。。。
  • Mybatis行为配置之Ⅰ—缓存
  • Linux系统安装MySQL
  • 泛型擦除到底是怎么一回事
  • 福建科立讯通信 指挥调度管理平台 多处文件上传漏洞复现