保存为npy格式并加载
import tensorflow as tf
import numpy as np
# 线性回归模型,别的模型保存方式也一样
x = tf.random_normal([100,1],mean=1.75,stddev=0.5,name="my_data")
y_true = tf.matmul(x,[[0.7]])+0.8
weight = tf.Variable(tf.random_normal([1,1],mean=0.1,stddev=1.0),name="weight")
bias = tf.Variable(0.1,name="bias")
y_predict = tf.matmul(x,weight)+bias
loss = tf.reduce_mean(tf.square(y_true-y_predict))
train_op=tf.train.GradientDescentOptimizer(0.1).minimize(loss)
init_op=tf.global_variables_initializer()
# 模型训练
with tf.Session() as sess:
#初始化变量
sess.run(init_op)
#打印随机初始化的权重和偏置
#print("随机初始化的权重参数为:%f,偏置为:%f" %(weight.eval(),bias.eval()))
#循环训练,运行优化
for i in range(50):
sess.run(train_op)
print("第%d次训练权重参数为:%