"HelloWorld" of Tensorflow

用Tensorflow实现最简单的神经网络——一个神经元
模拟一个最简单的线性结构

import tensorflow as tf
import numpy as np

#创建数据
xs = np.array([-1.0,0.0,1.0,2.0,3.0,4.0],dtype=float)
ys = np.array([-3.0,-1.0,1.0,3.0,5.0,7.0],dtype=float)

#创建模型
model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Dense(units=1,input_shape=[1]))
model.compile(optimizer="sgd",loss="man_squared_error")

#训练模型
model.fit(xs,ys,epochs=500) #因为数据很少，可以让epoch多一些

#预测模型
model.predict([10.0])