跑了下吴恩达大佬课上的一段代码:
我抄下来是这样的:
# TensorFlow例子:
# 假设J(w)是一个二次函数,同时是某个网络的损失函数
# 用TensorFlow来最小化这个损失函数,模拟神经网络的训练过程
# 流程:
# 定义参数 -> 定义损失函数 -> 定义训练(优化器、训练目标) -> 定义参数初始化器 ->
# 开启一个TensorFlow的session -> 初始化全局变量 -> 训练
import numpy as np
# import tensorflow as tf
w = tf.Variable(0, dtype=tf.float32)# 定义参数
cost = w ** 2 - 10 * w + 25# 定义损失函数
train = tf.compat.v1.train