当前位置:网站首页>TF通过feature与label生成(特征,标签)集合,tf.data.Dataset.from_tensor_slices

TF通过feature与label生成(特征,标签)集合,tf.data.Dataset.from_tensor_slices

2022-08-11 06:45:00 phac123

Numpy和Tensor格式都可以用这个语句读入数据

data = tf.data.Dataset.from_tensor_slices((输入特征, 标签))

代码

import tensorflow as tf

features = tf.constant([12, 23, 10, 17])
labels = tf.constant([0, 1, 1, 0])
dataset = tf.data.Dataset.from_tensor_slices((features, labels))
print(dataset)
for element in dataset:
    print(element)

在这里插入图片描述

原网站

版权声明
本文为[phac123]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_42596275/article/details/126259076