Skip to main content
 首页 » 编程设计

python之Tensorflow Estimator API 输入张量名称

2025年01月19日33telwanggs

在 tensorflow 中使用估算器并使用 tf.estimator.inputs.numpy_input_function() 传递输入时,为特征和标签输入创建的张量的名称是什么。

如果我打印出图表中所有占位符的名称,我会得到以下信息: name: "enqueue_input/Placeholder" name: "enqueue_input/Placeholder_1" name: "enqueue_input/Placeholder_2"

但是,这些张量的形状没有指定,所以我无法分辨哪个是哪个或为什么有 3 个张量,而不是只有一个特征和一个标签张量。

我意识到这里也有人问过这个问题:TensorFlow: What are the input nodes for tf.Estimator models

但是没有人回答。

请您参考如下方法:

numpy_input_fn 有两个重要参数:x 和 y。 x 是一个字典,它将特征列的名称与包含特征数据的数组相匹配。 y 是一个数组,其中包含 x 中的特征标签。

例如,以下代码将名为 x_coord 的特征在值为 0.5 时与标签 1 关联,在值为 1.2 时与标签 2 关联:

train_input = tf.estimator.inputs.numpy_input_fn( 
    x={"x_coord": np.array([0.5, 1.2])}, y=np.array([1, 2]))