首页 > 其他 > 详细

01_tf和numpy的区别

时间:2019-10-26 10:08:08      阅读:129      评论:0      收藏:0      [点我收藏+]


import numpy as np
import tensorflow as tf

# 这里是为了演示numpy和tf的区别。
np.random.seed(43)

x_data = np.random.rand(100).astype(np.float32)
y_data = x_data * 0.1 + 0.3
# w = np.random.rand()
# print(w)
# print(y_data)


# todo 2、 tf代码
# 一、构建模型图
with tf.Graph().as_default():
print(tf.get_default_graph())
w = tf.Variable(initial_value=tf.random_uniform([1], -1.0, 1.0))
b = tf.Variable(tf.zeros([1]))

y_hat = x_data * w + b
# print(y_hat)
loss = tf.reduce_mean(tf.square(y_data - y_hat))
print(loss)
optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.3)
train_opt = optimizer.minimize(loss)

epochs = 201
# 二、执行会话。
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
for e in range(epochs):
_, train_loss = sess.run([train_opt, loss])
if e % 30 == 0:
print(‘Epoch:{} - Train Loss:{}‘.format(e, train_loss))

01_tf和numpy的区别

原文:https://www.cnblogs.com/lvdongjie/p/11741744.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!