Tensorflow2.x框架-神经网络八股扩展-参数提取

参数提取，把参数存入文本

博主微信公众号（左）、Python+智能大数据+AI学习交流群（右）：欢迎关注和加群，大家一起学习交流，共同进步！

目录

摘要

一、提取可训练参数

二、设置 print 输出格式

三、完整代码

---

摘要

参数提取，把参数存入文本。

一、提取可训练参数

model.trainable_variables 返回模型中可训练的参数

二、设置 print 输出格式

np.set_printoptions(precision=小数点后按四舍五入保留几位，threshold=数组元素数量少于或等于门槛值，打印全部元素；否则打印门槛值+1个元素，中间用省略号补充)

注：precision=np.inf    # 打印小数位数；threshold=np.nan    # 打印全部数组元素

三、完整代码

"""
参数提取，把参数存入文本
"""
# 模块导入
import os
import numpy as np
import tensorflow as tf

# 设置显示全部内容，np.inf 表示无穷大
np.set_printoptions(threshold=np.inf)

# 导入数据集，分别为输入特征和标签
mnist = tf.keras.datasets.mnist
# (x_tra