欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

在MNIST中,两种方法将x(浮点张量)限制在0~1的范围内

程序员文章站 2022-05-05 08:57:35
...

这里介绍两种常用的方法

def load_data():
#加载MNIST数据集
    (x,y),(x_val,y_val)=datasets.minst.load_data()
    #转换为浮点张量,并缩放到-1~1
    x = tf.convert_to_tensor(x,dtype=tf.float32)/255.
    #转换为整形张量
    y = tf.convert_to_tensor(y,dtype=tf.int32)
    #one_hot编码
    y = tf.one_hot(y,depth=10)

方法2:

def preprocess(x,y):
    x = tf.cast(x,dtype=tf.float32)/255.
    y = tf.cast(y.dtype=tf.int32)
    return x,y

 

相关标签: 预处理