首页 > 解决方案 > 解析tensorflow/keras数据集时如何获取单个值?

问题描述

解析 tf 数据集时,您可以使用类似

def parse_example(x):
    return parse_example(
                x,
                {feature: tf.io.FixedLenFeature([], tf.string)
                 for feature in features
                 }
            )

这将返回一个字典(例如,dic={a:1, b:2, c:3}但是,有没有办法获得字典的单个值(例如,dic[a] = 1)?

标签: tensorflowkeras

解决方案


如果 'a' 是 dict 中的键,您可以通过将其用引号括起来来检索它的值:

dic["a"]


推荐阅读