回归问题

import tensorflow as tf
import numpy as np

# loss 的值
def compute_error_for_line_given_points(b, w, points):  # points 是一个两维数组
    totalError = 0
    for i in range(0, len(points)):
        x = points[i, 0]
        y = points[i, 1]
        totalError += (y - (w * x + b)) ** 2

    return totalError / float(len(points))


def step_gradient(b_current, w_current, points, learningRate):
    b_gradient = 0
    w_gradient = 0
    N = float(len(points))
    for i in range(0, len(points)):
        x = points[i, 0]
        y = points[i, 1]

        b_gradient += (2 / N) * ((w_current * x + b_current) - y)
        w_gradient += (2 / N) * x * ((w_current * x + b_current) - y)
    # update
    new_b = b_current - (learningRate * b_gradient)
    new_w = w_current - (learningRate * w_gradient)
    return [new_b,new_w]


def gradient_descent_runner(points,starting_b,staring_w,learning_rate,num_iterations):
    b = starting_b
    w = staring_w
    for i in range(num_iterations):
        b,w = step_gradient(b,w,np.array(points),learning_rate)
    return [b,w]

def run():
    points = np.genfromtxt("d:/data.csv",delimiter=‘,‘)
    print(points)
    learning_rate = 0.0001
    initial_b = 0
    initial_w = 0
    num_iterations = 1000
    print(f"Starting gradient descent at w:{initial_w},b:{initial_b},error={compute_error_for_line_given_points(initial_b,initial_w,points)}")
    print("Running...")


    [b,w] = gradient_descent_runner(points,initial_b,initial_w,learning_rate,num_iterations)

    print(f"After {num_iterations} iterations   w:{w},b:{b},error={compute_error_for_line_given_points(b,w,points)}")
    print("Running...")

if __name__ == ‘__main__‘:
    run()

import tensorflow as tf

from tensorflow.keras import datasets,layers,optimizers
import os
os.environ[‘TF_CPP_MIN_LOG_LEVEL‘] = ‘2‘

(x,y),(x_val,y_val) = datasets.mnist.load_data()
x = tf.convert_to_tensor(x,dtype=tf.float32)/255.
y = tf.convert_to_tensor(y,dtype=tf.int32)

y = tf.one_hot(y,depth=10)
print(x.shape,y.shape)
train_dataset = tf.data.Dataset.from_tensor_slices((x,y))
train_dataset = train_dataset.batch(200)


model = tf.keras.Sequential([
    layers.Dense(512,activation="relu"),
    layers.Dense(216, activation="relu"),
    layers.Dense(10),
])

optimizer = optimizers.SGD(learning_rate=0.001)
def train_epoch(epoch):
    for step,(x,y) in enumerate(train_dataset): # 300次循环
        with tf.GradientTape() as tape:
            x = tf.reshape(x,(-1,28*28))
            out = model(x)
            loss = tf.reduce_sum(tf.square(out - y))/ x.shape[0]

        grads = tape.gradient(loss,model.trainable_variables)
        optimizer.apply_gradients(zip(grads,model.trainable_variables))

        if step % 100 == 0:
            print(epoch,step,"loss:",loss.numpy())


def train():
    # 对数据集迭代30次
    for epoch in range(30):
        train_epoch(epoch)


if __name__ == ‘__main__‘:
    train()