Tensorflow入门基本教程

最近因为工作需要,需要将Tensorflow训练好的模型用C++接口调用,因此写了一个很小的demo来进行验证,在此过程中重复查找了不少资料,感觉挺浪费时间的,因此将这个demo记录下来。

生成Tensorflow模型文件

假设,你已经对Tensorflow的基本概念有了一定的了解,但是确不知道从何下手,下面给出一个拟合线性模型的完整代码,让你对Tensorflow有初步的印象。这段很短python代码先生成了一些二维数据,再通过线性变换,然后用随机梯度下降算法来拟合这些参数。

# -*- coding:utf-8 -*-
import tensorflow as tf
import numpy as np

ckpt_dir = './model'
if tf.gfile.IsDirectory(ckpt_dir) == False:
    tf.gfile.MakeDirs(ckpt_dir)
    print ('create directory: %s'%ckpt_dir)
# 使用 NumPy 生成假数据(phony data), 总共 100 个点.
x_data = np.float32(np.random.rand(100, 2))  # 随机输入
y_data = np.dot(x_data, [[0.100], [0.200]])

# 构造一个线性模型
x = tf.placeholder(tf.float32, shape=[None, 2], name='input')
W = tf.Variable(tf.random_uniform(shape=[2, 1]), name='weight')
y = tf.matmul(x, W, name='output')

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值