参数提取,把参数存入文本
博主微信公众号(左)、Python+智能大数据+AI学习交流群(右):欢迎关注和加群,大家一起学习交流,共同进步!
目录
摘要
参数提取,把参数存入文本。
一、提取可训练参数
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