一、sklearn模型保存与读取 
1、保存

1 from sklearn.externals import joblib
2 from sklearn import svm
3 X = [[0, 0], [1, 1]]
4 y = [0, 1]
5 clf = svm.SVC()
6 clf.fit(X, y)
7 joblib.dump(clf, "train_model.m")

2、读取

1 clf = joblib.load("train_model.m")
2 clf.predit([0,0]) #此处test_X为特征集

二、TensorFlow模型保存与读取(该方式tensorflow只能保存变量而不是保存整个网络,所以在提取模型时,我们还需要重新第一网络结构。) 
1、保存

 1 import tensorflow as tf
 2 import numpy as np
 3
 4 W = tf.Variable([[1,1,1],[2,2,2]],dtype = tf.float32,name='w')
 5 b = tf.Variable([[0,1,2]],dtype = tf.float32,name='b')
 6
 7 init = tf.initialize_all_variables()
 8 saver = tf.train.Saver()
 9 with tf.Session() as sess:
10          sess.run(init)
11          save_path = saver.save(sess,"save/model.ckpt")  

2、加载

1 import tensorflow as tf
2 import numpy as np
3
4 W = tf.Variable(tf.truncated_normal(shape=(2,3)),dtype = tf.float32,name='w')
5 b = tf.Variable(tf.truncated_normal(shape=(1,3)),dtype = tf.float32,name='b')
6
7 saver = tf.train.Saver()
8 with tf.Session() as sess:
9       saver.restore(sess,"save/model.ckpt")  

三、TensorFlow模型保存与读取(该方式tensorflow保存整个网络) 
1、保存

 1 import tensorflow as tf
 2
 3 # First, you design your mathematical operations
 4 # We are the default graph scope
 5
 6 # Let's design a variable
 7 v1 = tf.Variable(1. , name="v1")
 8 v2 = tf.Variable(2. , name="v2")
 9 # Let's design an operation
10 a = tf.add(v1, v2)
11
12 # Let's create a Saver object
13 # By default, the Saver handles every Variables related to the default graph
14 all_saver = tf.train.Saver()
15 # But you can precise which vars you want to save under which name
16 v2_saver = tf.train.Saver({"v2": v2})
17
18 # By default the Session handles the default graph and all its included variables
19 with tf.Session() as sess:
20   # Init v and v2
21   sess.run(tf.global_variables_initializer())
22   # Now v1 holds the value 1.0 and v2 holds the value 2.0
23   # We can now save all those values
24   all_saver.save(sess, 'data.chkp')
25   # or saves only v2
26   v2_saver.save(sess, 'data-v2.chkp')
27 模型的权重是保存在 .chkp 文件中,模型的图是保存在 .chkp.meta 文件中。

2、加载

 1 import tensorflow as tf
 2
 3 # Let's laod a previous meta graph in the current graph in use: usually the default graph
 4 # This actions returns a Saver
 5 saver = tf.train.import_meta_graph('results/model.ckpt-1000.meta')
 6
 7 # We can now access the default graph where all our metadata has been loaded
 8 graph = tf.get_default_graph()
 9
10 # Finally we can retrieve tensors, operations, etc.
11 global_step_tensor = graph.get_tensor_by_name('loss/global_step:0')
12 train_op = graph.get_operation_by_name('loss/train_op')
13 hyperparameters = tf.get_collection('hyperparameters')
14
15 恢复权重
16
17 请记住,在实际的环境中,真实的权重只能存在于一个会话中。也就是说,restore 这个操作必须在一个会话中启动,然后将数据权重导入到图中。理解恢复操作的最好方法是将它简单的看做是一种数据初始化操作。
18 with tf.Session() as sess:
19     # To initialize values with saved data
20     saver.restore(sess, 'results/model.ckpt-1000-00000-of-00001')
21     print(sess.run(global_step_tensor)) # returns 1000

四、keras模型保存和加载

1 model.save('my_model.h5')
2 model = load_model('my_model.h5') 

转载于:https://www.cnblogs.com/tectal/p/9053205.html

Sklearn,TensorFlow,keras模型保存与读取相关推荐

  1. Tensorflow2 图像分类-Flowers数据深度学习模型保存、读取、参数查看和图像预测

    目录 1.原文完整代码 1.1 模型运行参数总结 1.2模型训练效果 ​编辑2.模型的保存 3.读取模型model 4.使用模型进行图片预测 5.补充 如何查看保存模型参数 5.1 model_wei ...

  2. tensorflow 的模型保存和调用

    我们通常采用tensorflow来训练,训练完之后应当保存模型,即保存模型的记忆(权重和偏置),这样就可以来进行人脸识别或语音识别了. 1.模型的保存 # 声明两个变量 v1 = tf.Variabl ...

  3. 『TensorFlow』模型保存和载入方法汇总

    一.TensorFlow常规模型加载方法 保存模型 tf.train.Saver()类,.save(sess, ckpt文件目录)方法 参数名称 功能说明 默认值 var_list Saver中存储变 ...

  4. keras模型保存和加载

    (一)保存和加载整个模型 ​ 包含模型的结构.权重.训练配置项(损失函数.优化器).优化器状态,允许准确地从上次结束的地方开始训练. 1.训练完模型后 path='.../.../xxx.h5' mo ...

  5. tensorflow keras模型加载错误

    加载resnet101模型报错 mynet = tf.keras.applications.ResNet101(include_top=False, weights="imagenet&qu ...

  6. Tensorflow:如何保存/恢复模型?

    在Tensorflow中训练模型后: 您如何保存经过训练的模型? 您以后如何还原此保存的模型? #1楼 对于TensorFlow版本<0.11.0RC1: 保存的检查点包含模型中Variable ...

  7. keras/tensorflow 模型保存后重新加载准确率为0 model.save and load giving different result

    我在用别人的代码跑程序的时候遇到了这个问题: keras 模型保存后重新加载准确率为0 GitHub上有个issue:model.save and load giving different resu ...

  8. sklearn与pytorch模型的保存与读取

    当我们花了很长时间训练了一个模型,需要用该模型做其他事情(比如迁移学习),或者我们想把自己的机器学习模型分享出去的时候,我们这时候需要将我们的ML模型持久化到硬盘中去. 1.sklearn中模型的保存 ...

  9. keras中的模型保存和加载

    tensorflow中的模型常常是protobuf格式,这种格式既可以是二进制也可以是文本.keras模型保存和加载与tensorflow不同,keras中的模型保存和加载往往是保存成hdf5格式. ...

最新文章

  1. kali linux之edb--CrossFire缓冲区溢出
  2. GetAdaptersInfo获取MAC地址
  3. 计算机网络(二)-性能指标
  4. matlab hopty,运行Matlab时出现错误?Attempted to access rxd(500); index out of bou
  5. 【Vegas原创】windows2008配置、安装Exchange2007证书(for OWA,RPC over HTTP)
  6. Android使用PDFviewer渲染PDF文件
  7. 从0开始,使用豆瓣数据集做一个基于FM和逻辑回归的电影推荐系统
  8. UI实战教程之切图标注篇(UI必备)
  9. 推荐一款限时下载应用(听心字典)
  10. rpa项目poc_RPA POC概述
  11. 国际知名财务和ERP软件系统
  12. 把模糊MATLAB波形的图变清晰,什么修图软件能把模糊图片清晰化?
  13. 拓嘉辰丰电商:拼多多新上架的新品要不要做
  14. 使用Python读取文件夹中的图片
  15. dakai微信小程序 ios_苹果6s微信小程序进不去 苹果小程序怎么弄出来
  16. seo优化大纲之算法规则
  17. 【C语言】职工管理系统详解(文件操作)
  18. 创建LVM逻辑卷并挂载
  19. python——借书题
  20. 平板电脑html编程软件,HTML Kit Tools官方版

热门文章

  1. rk android8.1加密,Android 8.1RK平台增加自定义脚本,修改文件权限
  2. 远程定位gps linux,GPS(NMEA)数据解析
  3. c语言教程常用32,几个常用运算符
  4. java 爬虫框架_不知道Python爬虫?这篇文章丢给他(内含框架结构)
  5. 机器视觉:工业线阵相机与面阵相机特点分析
  6. 全球及中国公路行业运营模式与十四五价值前景分析报告2022版
  7. conn.execute参数
  8. 花生增产对话万书波-农业大健康·万祥军:获山东科技最高奖
  9. 《JavaScript 标准参考教程》阮一峰
  10. Android读取短信和联系人