美文网首页TensorFlow参考资料2018~2019
2018-06-25《TensorFlow模型保存、提取、预测》

2018-06-25《TensorFlow模型保存、提取、预测》

作者: 布口袋_天晴了 | 来源:发表于2018-06-25 21:39 被阅读21次

    任务一:写一个predict函数

    1.在模型初始化的地方添加了input和target命名空间:

    自定义命名空间

    2在模型初始化的地方添加了input和target命名空间,得到的TensorBoard可视化模型如下图:

    自定义命名空间input和target得到的可视化图

    3.TensorFlow模型保存和提取方法

    TensorFlow通过tf.train.Saver类实现神经网络模型的保存和提取。tf.train.Saver对象saver的save方法将TensorFlow模型保存到指定路径中,如保存在当前工作路径的model文件夹下saver.save(sess,'model/model.ckpt'),实际在这个文件目录下会生成4个文件:

    tensorflow模型保存,以及生成的4个文件

    注意:checkpoint文件是文本文件,可以查看其内容;其他三个文件都是二进制文件。

    1)checkpoint文件:主要保存的是模型文件列表。

    2)model.ckpt.meta二进制文件:保存了TensorFlow计算图的结构信息。

    3)model.ckpt.index二进制文件:保存每个变量的索引

    4)model.ckpt.data-00000-of-00001二进制文件:保存的是每个变量的权值

    模型提取(加载):

    使用该段代码,可以完成模型的加载

    模型预测:

    预测函数前半部分 预测函数后半部分

    参考文章:

    【1】TensorFlow模型保存和提取方法

    任务二:找词嵌入,别人训练好的

    资源 | Chinese Word Vectors:目前最全的中文预训练词向量集合

    项目链接:https://github.com/Embedding/Chinese-Word-Vectors

    已经训练好的,一个字就是一个向量的文件需要在这里面,下载了查看,看是不是符合自己的要求。

    任务三:在git上,用自己的名字全拼创建自己的代码分支

    1.git安装教程:Git安装和使用

    2.安装完成后,参考文章:git: windows git ssh keys生成

    生成key的过程 key放在id_rsa.pub文件内 生成ssh key,用于通过命令行获取和提交代码

    3.克隆仓库,参考文章:gitlab上克隆远程分支到本地

    仓库克隆到了我的E盘 克隆的仓库位置

    4.创建自己的分支

    问题1:fatal: not a git repository (or any of the parent directories): .git

    产生原因:一般是没有初始化git本地版本管理仓库,所以无法执行git命令

    解决方法:操作之前执行以下命令行: git init

    问题 解决方法

    问题2:fatal: Not a valid object name: 'master'.

    还未解决!!!!可参考文章:Git新建分支出现fatal: Not a valid object name: 'master'错误

    问题

    相关文章

      网友评论

        本文标题:2018-06-25《TensorFlow模型保存、提取、预测》

        本文链接:https://www.haomeiwen.com/subject/vtqdyftx.html