2021-01-04

学习目标：

tensorflow简单模型

---

学习产出：

#@Author    :xuxxx
#@Time      :2021/01/04
#@Software  :pycharm/tensorflow2.0/python3.8
#@Notes     :使用tensorflow实现已给简单的一元函数的线性模型

import tensorflow as tf
import numpy as np

input_xs = np.random.rand(1000)
input_ys = 3 * input_xs + 0.217
weight = tf.Variable(1.,dtype = tf.float32,name='weight')
bias = tf.Variable(1.,dtype=tf.float32,name='bias')

opt = tf.optimizers.Adam(1e-1)
def model(xs):
    logits = tf.multiply(xs,weight) + bias
    return logits

for xs,ys in zip(input_xs,input_ys):
    xs = np.reshape(xs,[1])
    ys = np.reshape(ys,[1])
    _loss = lambda :tf.losses.MeanSquaredError()(model(xs),ys)
    opt.minimize(_loss,[weight,bias])
    print(_loss().numpy)
print(weight)
print(bias)