tf.estimator.inputs.pandas_input_fn标签张量

时间:2018-06-28 11:17:44

标签: pandas tensorflow input

尝试使用我命名为pandas_input_fn()的pandas数据框来尝试Tensorflow内置的training_examples

这是一个非常简单的数据框,描述了一组功能和标签;然后将其作为参数传递到x中,作为参数pandas_input_fn(),如下所示,如果我正确理解了文档,应该返回一个输入函数,其中已将数据解析为特征和标签?

input_function = tf.estimator.inputs.pandas_input_fn(
x= training_examples,
y= None,
batch_size=128,
num_epochs=1,
shuffle=True,
queue_capacity=1000,
num_threads=1,
target_column='y'
)

但是,当我尝试将此函数传递给.train()方法时,出现如下所示的错误:

ValueError: You must provide a labels Tensor. Given: None. Suggested 
troubleshooting steps: Check that your data contain your label feature. Check 
that your input_fn properly parses and returns labels.

不确定我在做什么错吗?

1 个答案:

答案 0 :(得分:0)

尝试target_column = None并使用Y = training_examples ['label / target']中的实际Y列