小编AI-*_*ner的帖子

如何加载 Tensorflow 数据集“Iris”并将标签更改为单热编码

我正在尝试直接从 tensorflow 数据集加载“iris”数据集,但我被卡住了。我习惯于使用 CSV。

import tensorflow as tf
import tensorflow_datasets as tfds

data = tfds.load("iris",split='train[:80%]', as_supervised=True)
data = data.batch(10)
features, labels = data
Run Code Online (Sandbox Code Playgroud)

我不知道我应该如何分离特征 X,y。标签与特征的张量不同,但我不知道如何访问它们以使用它们。我想对标签进行热编码并将它们输入模型,但我被困在这里。

tensorflow 文档很少提供有关如何执行此操作的信息。任何帮助深表感谢

python tensorflow tensorflow-datasets iris-dataset tensorflow2.0

3
推荐指数
1
解决办法
1235
查看次数